args.resume error in type. SHould be str and should take path of model checkpoint

sacmehta / EdgeNets

This repository contains the source code of our work on designing efficient CNNs for computer vision

MIT License

411 stars 82 forks source link

args.resume error in type. SHould be str and should take path of model checkpoint #42

Open saiabinesh opened 3 years ago

saiabinesh commented 3 years ago

The following are the correct lines in train_detection.py to enable resume from checkpoint.

        checkpoint = torch.load(args.resume, map_location=torch.device('cpu'))
parser.add_argument('--resume', type=str, help='resume from checkpoint') #action='store_true',