Could you please guide me on how to resume interrupted training?

[limengran98]

Dear Author,

Thank you for your excellent work. Could you please guide me on how to resume interrupted training? The provided resume functionality in the code seems not to be working.

Best regards,

[peterant330]

Thanks for pointing out. We will fix it soon. You can refer to the file baselines/train.py first.

ckpt = torch.load(os.path.join(args.snapshot_path, "best.pth.tar"))
start_epoch = ckpt["epoch"]
best_loss = ckpt["best_val_loss"]
seg_net.load_state_dict(ckpt["network_dict"])
seg_net_opt.load_state_dict(ckpt["opt_dict"])
lr_scheduler.load_state_dict(ckpt["lr_scheduler_dict"])