@vmirly1 Just I forgot to ask about loading. I could see in here 49 loaded the epoch like this
checkpoint = torch.load(PATH)
model.load_state_dict(checkpoint['model_state_dict'])
optimizer.load_state_dict(checkpoint['optimizer_state_dict'])
epoch = checkpoint['epoch']
loss = checkpoint['loss']

could you please give me an example how can I load the model in specific epoch?  for example after first 10 epochs that saved.