about the checkpoint load

[Usernamezhx]

hi. thanks for your work. when i load the model you provide. it will show me that:ModuleNotFoundError: No module named 'torch'. My torch version is 1.5.0. but when I try torch.jit.load('face-attributes_scripted.pt') it will work. does it look right?Besides. how many epoch do you train ? I train with the default value you use except the batch size. I set 64 two graphics card: this is the train log: