search

amdegroot / ssd.pytorch

A PyTorch Implementation of Single Shot MultiBox Detector
MIT License
5.12k stars 1.74k forks source link

Transfer learning? #268

Open santhoshav15 opened 5 years ago

santhoshav15 commented 5 years ago

There is a nice tutorial in Keras to sample weights of the final layer in case we have one more of the desired classes we want to pick

Anyone can help me figure out how to do the same in pytorch?

krxat commented 5 years ago

There is this tutorial in the pytorch official website. This should help.