EdgeNets icon indicating copy to clipboard operation
EdgeNets copied to clipboard

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

Open saiabinesh opened this issue 4 years ago • 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',

saiabinesh avatar Nov 10 '21 20:11 saiabinesh