ShuweiShao / AF-SfMLearner

[MedIA2022 & ICRA2021] Self-Supervised Monocular Depth and Ego-Motion Estimation in Endoscopy: Appearance Flow to the Rescue
MIT License
101 stars 15 forks source link

关于预训练问题 #55

Closed evanismli closed 5 months ago

evanismli commented 7 months ago

您好,如果我想将SCARED数据集上的训练结果(仅需要解码器encoder)作为新一轮训练的权重加载进去,是使用--models_to_load吗?

ShuweiShao commented 5 months ago

--models_to_load 应该会把encoder和decoder的所有权重都加载进去,需要自己修改下。