我正在 Keras 中构建一个 NN,并且正在使用 Theano 后端。
代码:
cost = ((pred - target)**2).mean()
opt = RMSprop(lr = self.lr, rho = self.rho, epsilon = self.rms_epsilon)
#model -> keras nn
params = self.model.trainable_weights
updates = opt.get_updates(params, [], cost)
self.model.compile(optimizer = opt,loss='mse')
self._train = K.function([S, NS, A, R, T], cost, updates=updates)
如果我使用上述参数编译这个模型并保存它,加载的模型会保留训练配置(损失,优化)吗?
由于我使用 K.function(...) 方法来计算成本(损失),这会以任何方式影响模型的配置吗?
到目前为止,我还没有找到一种方法来检查恢复模型的配置是否与保存的配置相同。那么,有没有办法打印出网络配置以检查它是否已正确恢复,即使用与保存时完全相同的配置和参数?