在保存模型时加入以下代码忽略警告:
import warnings
import h5py
warnings.filterwarnings("ignore")
model.save('my_model.h5')
其中,warnings.filterwarnings("ignore")
用于忽略警告,model.save('my_model.h5')
用于保存模型。
上一篇:保存模型权重时出现“<tensor> 不是此图的元素。”
下一篇:保存模型时出现“NameError: name 'FileCheckpointManager' is not defined”错误。