lvpengyuan/corner

Issues related to training the model

Opened this issue · 0 comments

I am testing your code. Accordingly,
i) I cloned the repository, executed the .sh file
ii) Downloaded the model and placed it in weights/ folder
iii) Tried to train the model using the command python train.py

I got the error as follows:

Traceback (most recent call last): File "train.py", line 11, in <module> from data import train_cfg, AnnotationTransform, ICDARDetection, detection_collate, SynthDetection, TD500Detection, MLTDetection File "/home/pg2017/cse/17071016/corner/data/__init__.py", line 1, in <module> from .icdar import ICDARDetection File "/home/pg2017/cse/17071016/corner/data/icdar.py", line 6, in <module> import torchvision.transforms as transforms File "/home/pg2017/cse/17071016/.conda/envs/myenv/lib/python3.7/site-packages/torchvision/__init__.py", line 1, in <module> from torchvision import models File "/home/pg2017/cse/17071016/.conda/envs/myenv/lib/python3.7/site-packages/torchvision/models/__init__.py", line 11, in <module> from . import detection File "/home/pg2017/cse/17071016/.conda/envs/myenv/lib/python3.7/site-packages/torchvision/models/detection/__init__.py", line 1, in <module> from .faster_rcnn import * File "/home/pg2017/cse/17071016/.conda/envs/myenv/lib/python3.7/site-packages/torchvision/models/detection/faster_rcnn.py", line 7, in <module> from torchvision.ops import misc as misc_nn_ops File "/home/pg2017/cse/17071016/.conda/envs/myenv/lib/python3.7/site-packages/torchvision/ops/__init__.py", line 1, in <module> from .boxes import nms, box_iou File "/home/pg2017/cse/17071016/.conda/envs/myenv/lib/python3.7/site-packages/torchvision/ops/boxes.py", line 2, in <module> from torchvision import _C ImportError: /home/pg2017/cse/17071016/.conda/envs/myenv/lib/python3.7/site-packages/torchvision/_C.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZN3c1019UndefinedTensorImpl10_singletonE

What is the possible reason for the error? As I have never faced such an error.

Thanks in advance. Waiting for a reply.