ModelFeast icon indicating copy to clipboard operation
ModelFeast copied to clipboard

where to find every module's checkpoint_best.pth?

Open dongnana777 opened this issue 6 years ago • 2 comments

dongnana777 avatar May 21 '19 03:05 dongnana777

I am wondering if you could kindly send me the pretrained modules for 3D CNN and the necessary information about it. I promise they will be only used for research. Thank you very much for your kind consideration and I am looking forward to your early reply.

dongnana777 avatar May 21 '19 06:05 dongnana777

1、where to find .pth files All the checkpoint files are saved in this folder: ".saved", For example, if you train a resnet18 at "/home/haha/" in 13th February, 11:47am, you will find your files here: '/home/haha/saved/resnet18/0213_114719/checkpoint_best.pth'

Besides, you can change your checkpoint path by use this function: https://github.com/daili0015/ModelFeast/blob/master/classifier.py#L203 set_trainer(self, save_dir="saved/")

2、pretrained models I did not train all the models in my codes, but these 3D pretrained models may helps you: https://github.com/kenshohara/3D-ResNets-PyTorch

daili0015 avatar May 21 '19 14:05 daili0015