MingSun-Tse / Collaborative-Distillation

[CVPR'20] Collaborative Distillation for Ultra-Resolution Universal Style Transfer (PyTorch)
MIT License
185 stars 23 forks source link

No module named 'torch.utils.serialization #13

Closed ioannist closed 3 years ago

ioannist commented 4 years ago

UDA_VISIBLE_DEVICES=0 python WCT.py --debug --mode original -->

Traceback (most recent call last):
  File "WCT.py", line 10, in <module>
    from util_wct import WCT
  File "/home/ioannis/Desktop/Collaborative-Distillation/PytorchWCT/util_wct.py", line 3, in <module>
    from torch.utils.serialization import load_lua
ModuleNotFoundError: No module named 'torch.utils.serialization'
ioannist commented 4 years ago

Installed the dependencies form requirements file and it works. Had to downgrade python from 3.8 to 3.6.9 too in the conda env.

inmchoi commented 4 years ago

when pytorch updated 0.4 to 1.1, they got rid of all legacy stuff like serialization so if you use 0.4 version, it will be solved

MingSun-Tse commented 3 years ago

At this point, I don't have more practical suggestions about how to run our code on environments other than that we suggested. So I am closing this issue. If anyone still has problem with it, please let me know.