怎么修改别人的预训练模型

怎么修改别人预训练好的图像压缩模型呢

刚接触深度学习，这是中科大的一篇端到端图像压缩的代码。我只会加载了别人预训练好的model然后运行main.py，得到结果之后要怎么修改呢？
本人做的图像水印方向，我导师说想修改这个模型 用在图像水印的过程中。真的完全不懂，有没有了解的人能帮忙指导一下

img

请问你要修改什么呢？