-
Notifications
You must be signed in to change notification settings - Fork 58
Description
Traceback (most recent call last):
File "/home/dyk/anaconda3/envs/hawp/lib/python3.9/runpy.py", line 197, in _run_module_as_main
return _run_code(code, main_globals, None,
File "/home/dyk/anaconda3/envs/hawp/lib/python3.9/runpy.py", line 87, in _run_code
exec(code, run_globals)
File "/media/usv012/dyk/hawp/hawp/fsl/train.py", line 266, in
train(cfg, model, train_dataset, optimizer, scheduler, loss_reducer, checkpointer, arguments)
File "/media/usv012/dyk/hawp/hawp/fsl/train.py", line 114, in train
loss_dict, extra_info = model(images,annotations)
File "/home/dyk/anaconda3/envs/hawp/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1773, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/dyk/anaconda3/envs/hawp/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1784, in _call_impl
return forward_call(*args, **kwargs)
File "/media/usv012/dyk/hawp/hawp/fsl/model/models.py", line 215, in forward
return self.forward_train(images, annotations=annotations)
File "/media/usv012/dyk/hawp/hawp/fsl/model/models.py", line 518, in forward_train
targets , metas = self.hafm_encoder(annotations)
File "/media/usv012/dyk/hawp/hawp/fsl/model/hafm.py", line 18, in call
t,m = self._process_per_image(ann)
File "/media/usv012/dyk/hawp/hawp/fsl/model/hafm.py", line 73, in _process_per_image
lmap, _, _ = _C.encodels(lines,height,width,height,width,lines.size(0))
AttributeError: 'NoneType' object has no attribute 'encodels'
Anyone met the same error above? My installation follows README.md, only cuda version is different(my cuda version is 11.3).