steganogan 训练好的模型保存在哪
时间: 2023-12-27 10:01:53 浏览: 100
Pytorch——保存训练好的模型参数
5星 · 资源好评率100%
SteganoGAN训练好的模型默认保存在项目根目录下的`saved_models`文件夹中。具体来说,生成器和鉴别器的权重分别保存在`saved_models/generator_weights.h5`和`saved_models/discriminator_weights.h5`文件中。你也可以在训练脚本中通过更改`--model_path`参数来指定模型的保存路径。
阅读全文