CondenseNet icon indicating copy to clipboard operation
CondenseNet copied to clipboard

Transfer-learning with CondenseNet

Open BDimitri opened this issue 7 years ago • 1 comments

Hello, I am getting good accuracy with CondenseNet on my dataset when training from scratch, but I feel like I could boost the results if I could train from a checkpoint pretrained on ImageNet. You offer the converted checkpoints on ImageNet, but my understanding of the situation is that I can't use that for transfer learning because its missing some dicts like the optimizer. Since I am new to pytorch I feel like I am missing something. Is it possible to train from the converted checkpoint? If no, would it be possible to upload the unconverted model?

BDimitri avatar Jul 11 '18 13:07 BDimitri

Sorry for being late. As far as I know, you don't need the state dict of optimizer to do transfer learning. Just load the converted weight and train should be fine.

ShichenLiu avatar Sep 15 '18 17:09 ShichenLiu