Resume Training After Interruption

[mabubakarsaleem]

Hi,

I was training my model using the configuration file train_HAT_SRx4_finetune_from_ImageNet_pretrain.yml, and the training got interrupted. I have the model checkpoint net_g_1500.pth and the training state 1500.state.

Could you please guide me on how to update the configuration file below to resume training from this checkpoint and state?

File below : train_HAT_SRx4_finetune_from_ImageNet_pretrain.yml

path:
pretrain_network_g: ./experiments/train_HAT_SRx4_ImageNet_from_scratch/models/HAT_SRx4_ImageNet-pretrain.pth
param_key_g: 'params_ema'
strict_load_g: true
resume_state: ~

and if there are other settings which i should change then also guide me and also guide me the command to resume that training.

And also guide me if i want to save my weights in google drive during the training on goole colab then on which file i should change to save the models saving path.