pytorch/examples

OSError: [Errno 22] Invalid argument:

Open

#887 ouverte le 14 févr. 2021

Voir sur GitHub
 (3 commentaires) (0 réactions) (0 assignés)Python (21 634 stars) (9 429 forks)batch import
good first issue

Description

https://github.com/pytorch/examples/blob/36441a83b6595524a538e342594ee6482754f374/fast_neural_style/neural_style/neural_style.py#L117

Spent 2 hours training a model for it to fail the save with error prefixed with OSError: [Errno 22] Invalid argument :(

For now I'm just adding fail-safe but I am sure it can be written cleaner

custom_save_path = save_model_path
while True:
    try:
        torch.save(transformer.state_dict(), custom_save_path)
        break
    except:
        print("Please enter new save path:")
        custom_save_path = input() 

print("\nDone, trained model saved at", custom_save_path)

Guide contributeur