Deep-Feature-Flow icon indicating copy to clipboard operation
Deep-Feature-Flow copied to clipboard

is it a bug?

Open liurongdev opened this issue 7 years ago • 0 comments

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 model @YuwenXiong @ancientmooner @cddlyf @liaojing @liurongdev

liurongdev avatar Dec 25 '18 13:12 liurongdev