amdegroot / ssd.pytorch

A PyTorch Implementation of Single Shot MultiBox Detector
MIT License
5.14k stars 1.75k forks source link

continue train #495

Open czy112 opened 4 years ago

czy112 commented 4 years ago

train break how to contiune? Thanks

l1158984238 commented 4 years ago

image you can modify parser.add_argument --resume such as : parser.add_argument('--resume', default='/home/xtu07/ssd.pytorch-master/weights/ssd300_COCO_60000.pth', type=str, help='Checkpoint state_dict file to resume training from')

JOSHUAWENJZ commented 4 years ago

@czy112 I tried @l1158984238 's method. That is worked! @l1158984238 Thank you for your help!