search

ZhangHanbo / Visual-Manipulation-Relationship-Network-Pytorch

MIT License
41 stars 13 forks source link

ImportError: No module named ops #4

Open li-source opened 3 years ago

li-source commented 3 years ago

when i run python main.py --dataset vmrdcompv1 --frame mgn --net res101 --cuda

Traceback (most recent call last): File "/home/litong/Visual-Manipulation-Relationship-Network-Pytorch-pytorch1.0/main.py", line 25, in from model.utils.net_utils import weights_normal_init, save_net, load_net, \ File "/home/litong/Visual-Manipulation-Relationship-Network-Pytorch-pytorch1.0/model/utils/net_utils.py", line 11, in from model.roi_layers import nms File "/home/litong/Visual-Manipulation-Relationship-Network-Pytorch-pytorch1.0/model/roi_layers/init.py", line 3, in from .nms import nms File "/home/litong/Visual-Manipulation-Relationship-Network-Pytorch-pytorch1.0/model/roi_layers/nms.py", line 3, in from torchvision.ops import nms ImportError: No module named ops

ZhangHanbo commented 3 years ago

when i run python main.py --dataset vmrdcompv1 --frame mgn --net res101 --cuda

Traceback (most recent call last): File "/home/litong/Visual-Manipulation-Relationship-Network-Pytorch-pytorch1.0/main.py", line 25, in from model.utils.net_utils import weights_normal_init, save_net, load_net, File "/home/litong/Visual-Manipulation-Relationship-Network-Pytorch-pytorch1.0/model/utils/net_utils.py", line 11, in from model.roi_layers import nms File "/home/litong/Visual-Manipulation-Relationship-Network-Pytorch-pytorch1.0/model/roi_layers/init.py", line 3, in from .nms import nms File "/home/litong/Visual-Manipulation-Relationship-Network-Pytorch-pytorch1.0/model/roi_layers/nms.py", line 3, in from torchvision.ops import nms ImportError: No module named ops

If you use the pytorch1.0 branch, you should make sure that the torch version >= 1.4.0.