sacmehta/EdgeNets

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

Opened this issue · 0 comments

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',