如何在预训练模型的基础上实现迁移学习？

youyuge34 / Anime-InPainting

An application tool of edge-connect, which can do anime inpainting and drawing. 动漫人物图片自动修复，去马赛克，填补，去瑕疵

Other

1.08k stars 152 forks source link

Open dong0520 opened 3 years ago

dong0520 commented 3 years ago

之前使用自己的数据集训练EdgeConnect模型，同样注意到了过久的训练时长。读源代码时只看到在loss.py中冻结了VGG19的部分参数，没有找到迁移学习有关的冻结网络层、调用预训练模型中参数等代码。请问大佬具体是怎样实现的迁移学习？在项目代码中有写吗？感恩感恩~