Open Solitude1024 opened 3 months ago
作者你好,我有一些关于训练的问题,对于训练命令“./train.sh Enhancement/Options/WaveNet____.yml”:
(1)是不是从零开始的训练?即 是否加载预训练权重 (前提:我想稍微魔改一下网络试试)对于“修改网络后加载预训练权重训练”,根据我查到的方法,是将net.load_state_dict()中的strict设置为False,但是.load_state_dict()似乎只在demo.py中存在,train.py中没有找到。 (2)与(1)相关的是,以Enhancement/Options/WaveNet_B_LOL.yml为例,62-66行 #path中三个参数的含义我不太确定,如何设置呢?
A(1):你好,要fine tune的话 你可以直接在/basicsr/train.py文件166-188行修改为具体的预训练权重路径就行了。 A(2):具体参数以及更深层的定制化可以参考basicSR官方文档,https://github.com/XPixelGroup/BasicSR-docs
作者你好,我有一些关于训练的问题,对于训练命令“./train.sh Enhancement/Options/WaveNet**.yml”:
(1)是不是从零开始的训练?即 是否加载预训练权重 (前提:我想稍微魔改一下网络试试)对于“修改网络后加载预训练权重训练”,根据我查到的方法,是将net.load_state_dict()中的strict设置为False,但是.load_state_dict()似乎只在demo.py中存在,train.py中没有找到。 (2)与(1)相关的是,以Enhancement/Options/WaveNet_B_LOL.yml为例,62-66行 #path中三个参数的含义我不太确定,如何设置呢?