Deep-Feature-Flow
Deep-Feature-Flow copied to clipboard
is it a bug?
in the file train_end2end.py
if config.TRAIN.RESUME:
print('continue training from ', begin_epoch)
arg_params, aux_params = load_param(prefix, begin_epoch, convert=True)
but do_checkpoint is iter_no+1
def do_checkpoint(prefix, means, stds):
def _callback(iter_no, sym, arg, aux):
mx.model.save_checkpoint(prefix, iter_no + 1, sym, arg, aux)
return _callback
if config file train.resume==true the program cannot find the right model file so l change the code ,make begin_epoch+1
if config.TRAIN.RESUME:
print('continue training from ', begin_epoch)
arg_params, aux_params = load_param(prefix, begin_epoch+1, convert=True)
and it work right
Attachment as shown
@YuwenXiong @ancientmooner @cddlyf @liaojing @liurongdev