Open itouchzh opened 1 year ago
@itouchzh can you you tell what steps you carried out to start training and did you keep the dataset as it is for training or made some changes. Please Help
File "/content/TryYours-Virtual-Try-On/HR-VITON-main/train_condition.py", line 503, in
Hi. How can I resume training from the breakpoint without starting over when the training is interrupted unexpectedly? Is modifying the load_step option in the code and loading the desired checkpoint a correct approach? Like this:
parser.add_argument("--load_step", type=int, default=70000) parser.add_argument('--gen_checkpoint', type=str, default='./checkpoints/gen_step_070000.pth', help='gen checkpoint') parser.add_argument('--dis_checkpoint', type=str, default='./checkpoints/dis_step_070000.pth', help='dis checkpoint')
Is there anything else that needs to be done besides this? Thanks.