When saving a model like the following snip of code, does it then matter if the model is in train() or eval() mode? Does train() and eval() change the model.state_dict()? And if so, should model.train() be called before saving the model?
for epoch in range(5):
    # prepare model for training
    model.train()
    for i, (local_batch, local_labels) in enumerate(training_generator):
        ....

    # prepare model for evaluation
    model.eval()
    for i, (local_batch, local_labels) in enumerate(validation_generator):
        ....
    
    torch.save(model.state_dict(),  os.path.join('models', f'model_{epoch+1:03}.pt'))