PCR
PCR copied to clipboard
AttributeError: dyn_poly_cls
Hello. Thanks for your great work.
I organized the CTW1500 and ran ./train.sh, but got this error.
Traceback (most recent call last):
File "train_net.py", line 84, in <module>
main()
File "train_net.py", line 80, in main
train(cfg, network)
File "train_net.py", line 46, in train
trainer.train(epoch, train_loader, optimizer, recorder)
File "***/PCR/lib/train/trainers/trainer.py", line 38, in train
output, loss, loss_stats, image_stats = self.network(batch)
File "/home/usrs/***/miniconda3/envs/pt1.4cuda10.1/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in __call__
result = self.forward(*input, **kwargs)
File "/home/usrs/***/miniconda3/envs/pt1.4cuda10.1/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py", line 150, in forward
return self.module(*inputs[0], **kwargs[0])
File "/home/usrs/***/miniconda3/envs/pt1.4cuda10.1/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in __call__
result = self.forward(*input, **kwargs)
File "lib/train/trainers/snake.py", line 25, in forward
output = self.net(batch['inp'], batch)
File "/home/usrs/***/miniconda3/envs/pt1.4cuda10.1/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in __call__
result = self.forward(*input, **kwargs)
File "lib/networks/snake/ct_snake.py", line 38, in forward
output = self.gcn(output, cnn_feature, batch)
File "/home/usrs/***/miniconda3/envs/pt1.4cuda10.1/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in __call__
result = self.forward(*input, **kwargs)
File "lib/networks/snake/evolve.py", line 379, in forward
init['py_ind'], cfg.poly_cls_branch)
File "lib/networks/snake/evolve.py", line 258, in evolve_poly
evolve_polys, evolve_polys_cls = snake(init_input, adj)
File "/home/usrs/***/miniconda3/envs/pt1.4cuda10.1/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in __call__
result = self.forward(*input, **kwargs)
File "lib/networks/snake/snake.py", line 182, in forward
if cfg.dyn_poly_cls:
File "***/PCR/lib/config/yacs.py", line 115, in __getattr__
raise AttributeError(name)
AttributeError: dyn_poly_cls
When I viewed lib/networks/snake/snake.py, I found "cfg.dyn_poly_cls". So, I looked for dyn_poly_cls in lib/config/config.py, but cfg.dyn_poly_cls was not found.
Here is my environment information.
Ubuntu Server 18.04
CUDA: 10.1
Python: 3.7.11
pytorch: 1.4.0
torchvision: 0.5.0
cudatoolkit: 10.1.243
I have no idea how to fix this problem. Could you give me some help?
Thank you for reading.