keras 如何提取传递给TensorFlow函数的所有参数?

kuhbmx9i  于 2023-02-04  发布在  其他
关注(0)|答案(1)|浏览(111)

很难在新数据中重新训练模型,因为我从来不记得初始优化器、损失函数和超参数。如何提取传递给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)检查它,但是输出没有过滤。

tkclm6bt

tkclm6bt1#

我刚刚找到了一个方法。缺点是它需要先编译模型。我会发布它,因为也许有人有同样的问题。

model.optimizer.get_config()

相关问题