DeniJsonC / WaveNet

[PG 2023] WaveNet: Wave-Aware Image Enhancement
Other
78 stars 14 forks source link

训练设置 #12

Open Solitude1024 opened 3 months ago

Solitude1024 commented 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中三个参数的含义我不太确定,如何设置呢?

DeniJsonC commented 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