使用pytorch进行迁移学习 修改预训练权重模型的分类层/输出类别(附mmaction相关实现代码)

文章目录 example step1 文件创建 step2 得到预训练模型 step3 模型修改 参考文章 example 以mmaction中的tsn+resnet50+kinetics训练的模型做为预训练模型,拿来训练TSN+resnet+ucf101。python step1 文件创建 创建文件tools/transfer_model.py。git step2 得到预训练模型 在mmacti
相关文章
相关标签/搜索