saiabinesh
saiabinesh
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',
Full error Below usage: frcnn.py [-h] --run_name RUN_NAME [--ckpt CKPT] [--dataset {voc07,voc12,coco}] [--proposals {mcg,edgeboxes,}] --num_classes NUM_CLASSES [--extend EXTEND] [--num_layers NUM_LAYERS] --action ACTION [--data_format {NHWC,NCHW}] [--sigmoid] [--print_step PRINT_STEP] [--conf_thresh CONF_THRESH] [--nms_thresh...