search

VISION-SJTU / RECCE

[CVPR2022] End-to-End Reconstruction-Classification Learning for Face Forgery Detection
MIT License
116 stars 16 forks source link

关于预训练或加载权重训练的问题 #42

Open 0coldmoon0 opened 2 months ago

0coldmoon0 commented 2 months ago

您好，我想询问一下本代码支不支持加载预训练权重训练的方式，或者加载某次训练生成的.bin文件继续训练的方式。如果支持，应该修改代码或命令的哪个部分？

XJay18 commented 1 month ago

您好，您可以将config文件中的resume设置为True（训练过程中读取该值的位置在这里），然后在Trainer中实现一下_load_ckpt这个函数以加载对应权重继续训练。