how to continue training from restored checkpoint?

[kevinstan]

Hi @ashual, is it possible to continue training using the pretrained models you uploaded? It seems like the checkpoint_with_model.pt only works for the sample_images.py script. The checkpoint only contains 'model_state' but not the states of other modules for training, e.g. 'optim_state', 'd_obj_state', 'd_img_state', 'd_mask_state', etc.

Can you please upload the full checkpoints during training?

Thanks,
Kevin

[ashual]

unfortunately I don't have the original checkpoints anymore.
But I think continue training will just cause overfitting.