TF 2.0 Beta：如何重新初始化自定义模型的权重？

Question

因此，我有一个通过对模型和层进行子类化而构建的模型。我正在做一些实验，我想返回一个新初始化的NN。我怎样才能做到这一点？ 当前，我只是运行再次定义它的代码。应该有一个更整洁的方法。 我尝试了model.build（运行两次）。没用。

Answer 1

如果您要使用相同的随机权重进行重新初始化，则可以使用以下命令将预训练权重保存到磁盘中：

# build model, model.compile(), etc

model.save_weights('initial_weights.h5')

# model.fit(), etc

然后，当您想返回时，可以使用以下方法再次加载初始砝码：

model.load_weights('initial_weights.h5')