很难在新数据中重新训练模型,因为我从来不记得初始优化器、损失函数和超参数。如何提取传递给TensorFlow函数的所有参数?让我们从下面的代码中,举例说明如何提取包含learning_rate、beta_1、beta_2等参数的列表。
optimizer = tf.keras.optimizers.Adam(learning_rate=0.001,
beta_1=0.9,beta_2=0.999,
epsilon=1e-07, amsgrad=False,
name="Adam")
我只想提取名称,这样我以后就可以调用它们,例如:
optimizer.learning_rate
我试过.keys()
,.classes()
,但是没有任何效果。当然我可以用dir(optimizer)
检查它,但是输出没有过滤。
1条答案
按热度按时间tkclm6bt1#
我刚刚找到了一个方法。缺点是它需要先编译模型。我会发布它,因为也许有人有同样的问题。