deepinsight / deepinsight/insightface

How to train only fc7 layer of face recognition model and keeping pretrained weights frozen?

Open
#1,077 3 comments 4 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
29.7k
Forks
6.1k
PR merge metrics
No merged PRs in 30d

Description

I want to fine-tune res50 pretrained model on VGG2 dataset. The res50 is originally trained on MS1M. Fc7 layer's weights of res50 are not provided and only base network weights are available. I want to freeze pretrained weights and first only train classification layer for VGG2. Once classification loss is reduced then I want to train entire network. Can anyone help, how to update train.py script for freezing layers before fc7?

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.