共享SR和非共享TP

[zhaoguoqing12]

论文中多阶段训练 提出使用共享SR和非共享TP,但是代码中写的是 非共享SR和共享TP


根据你提出的训练命令
python3 -u main.py --arch="tsrn_tl_cascade" --batch_size=48 --STN --mask --use_distill --gradient --sr_share --stu_iter=3 --vis_dir='vis_TPGSR-TSRN'
--sr_share 默认为False,训练时是True

[vardenppp]

这里应该：上面是如果不sr_share，初始化多个sr模块，下面是初始化多个tpg模块。
但最后实际应用的时候，是根据是否share进行选择的，你可以搜一下“pick”，代码里写的是如果share，pick=0，否则pick=i。

[zhaoguoqing12]

@vardenppp 感谢,我看到了。所以训练不加 sr_share 并且 加上 tpg_share，才是共享SR和非共享TP。