在编译模型之后但在训练之前,保存初始权重:
model.save_weights('model.h5')
然后在训练后,通过重新加载初始权重来“重置”模型:
model.load_weights('model.h5')
这使您可以从Apple到Apple模型比较不同的数据集,并且比重新编译整个模型要快。
重置Keras层中的权重
在编译模型之后但在训练之前,保存初始权重:
model.save_weights('model.h5')
然后在训练后,通过重新加载初始权重来“重置”模型:
model.load_weights('model.h5')
这使您可以从Apple到Apple模型比较不同的数据集,并且比重新编译整个模型要快。