Hi, you can do smth like
torch.save({
        "epoch": epoch + 1,
        "model_state_dict": model.state_dict(),
        ...
        "optimizer": optimizer.state_dict(),
        "loss": loss
    }, model_path)

but specify only parameters you want to save