Anime-InPainting icon indicating copy to clipboard operation
Anime-InPainting copied to clipboard

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

Open dong0520 opened this issue 4 years ago • 0 comments

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

dong0520 avatar Oct 07 '20 06:10 dong0520