Add comment
This commit is contained in:
parent
87f59638dd
commit
0f419b37f6
@ -6,7 +6,8 @@ import numpy as np
|
|||||||
from keras_core.backend.config import backend
|
from keras_core.backend.config import backend
|
||||||
|
|
||||||
if backend() == "pytorch":
|
if backend() == "pytorch":
|
||||||
# TODO: comment explanation
|
# torch needs to be imported first, otherwise it will
|
||||||
|
# segfault upon import.
|
||||||
import torch
|
import torch
|
||||||
|
|
||||||
from keras_core.backend.common.keras_tensor import KerasTensor
|
from keras_core.backend.common.keras_tensor import KerasTensor
|
||||||
|
@ -1,4 +1,5 @@
|
|||||||
import torch
|
import torch
|
||||||
|
|
||||||
from keras_core.backend.pytorch.core import to_torch_dtype
|
from keras_core.backend.pytorch.core import to_torch_dtype
|
||||||
|
|
||||||
|
|
||||||
|
@ -9,4 +9,5 @@ pytest
|
|||||||
pandas
|
pandas
|
||||||
absl-py
|
absl-py
|
||||||
requests
|
requests
|
||||||
h5py
|
h5py
|
||||||
|
torch
|
Loading…
Reference in New Issue
Block a user