Finetuning - freeze_trunk

NVIDIA / semantic-segmentation

Nvidia Semantic Segmentation monorepo

BSD 3-Clause "New" or "Revised" License

1.76k stars 388 forks source link

Finetuning - freeze_trunk #157

Open joSilva99 opened 2 years ago

joSilva99 commented 2 years ago

Hi, please I need some help. I'm trying to finetuning using the --freeze_trunktag. But when I search in the code I can't find the usage of the tag. You didn't implemented this functionality? I can't find the place where you check if args.freeze_trunk is true. Can you give some tip?

Thanks

karansapra commented 2 years ago

hi,

Sorry, I think that is legacy code from our previous repo that we need to cleanup. You would need to manually insert torch.no_grad support for this. Our trunk is quite isolated so this should be rather easy if you just trying to freeze the trunk.