lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add comment

Browse Source
This commit is contained in:
Francois Chollet 2023-05-15 13:25:44 -07:00
parent 87f59638dd
commit 0f419b37f6
3 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
keras_core/backend/__init__.py
View File

@ -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
keras_core/backend/pytorch/numpy.py
View File

@ -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

1
requirements.txt
View File

@ -10,3 +10,4 @@ pandas
absl-py absl-py
requests requests
h5py h5py
torch