keras.optimizers.Adam 在 Colab 中运行 GPU 代码时"not found module"错误



我正在尝试运行代码导入keras。优化器亚当。我想从GPU做,因为我需要做一个长时间的训练。如果我正常运行,它可以工作,但每次我将运行时环境更改为GPU时,它都找不到模块"adam"。给出了下面的误差。由于与其他库的兼容性,我使用keras 2.2.4:

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
<ipython-input-12-ca97e5a6925a> in <module>()
8 from keras.models import Sequential
9 from keras.layers import Dense, Activation, Flatten, Convolution2D, Permute
---> 10 from keras.optimizers import Adam
11 import keras.backend as K
12 
ImportError: cannot import name 'Adam' from 'keras.optimizers' (/usr/local/lib/python3.7/dist-packages/keras/optimizers.py)

NOTE: If your import is failing due to a missing package, you can
manually install dependencies using either !pip or !apt.
To view examples of installing some common dependencies, click the
"Open Examples" button below.

换行

from keras.optimizers import Adam

:

from tensorflow.keras.optimizers import Adam

,它会工作的。

最新更新