【发布时间】:2019-04-01 15:53:10
【问题描述】:
对于我的强化学习应用程序,我需要能够应用自定义渐变/最小化不断变化的损失函数。根据documentation,应该可以使用 Optimizer.minimize() 函数。但是,我的 pip 安装版本似乎根本没有这个功能。
我的代码:
from tensorflow.python.keras.optimizers import Adam, SGD
print(tf.version.VERSION)
optim = Adam()
optim.minimize(loss, var_list=network.weights)
输出:
2.0.0-alpha0
Traceback (most recent call last):
File "/Users/ikkamens/Library/Preferences/PyCharmCE2018.3/scratches/testo.py", line 18, in <module>
optim.minimize(loss, var_list=network.weights)
AttributeError: 'Adam' object has no attribute 'minimize'
【问题讨论】:
-
应该是
tf.optimizers.Adam() -
根据文档,tf.optimizers 别名为 tf.keras.optimizers:tensorflow.org/versions/r2.0/api_docs/python/tf/optimizers/…