Giters

HRNet / HRNet-Semantic-Segmentation

The OCR approach is rephrased as Segmentation Transformer: https://arxiv.org/abs/1909.11065. This is an official implementation of semantic segmentation for HRNet. https://arxiv.org/abs/1908.07919

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

training error-> AttributeError: 'NoneType' object has no attribute 'copy'

liwei1101 opened this issue · comments

commented

=> init weights from normal distribution
=> init weights from normal distribution
Traceback (most recent call last):
Traceback (most recent call last):
File "tools/train.py", line 329, in
File "tools/train.py", line 329, in
main()
File "tools/train.py", line 294, in main
main()
File "tools/train.py", line 294, in main
trainloader, optimizer, model, writer_dict)
File "/public/home/liw/Code/HRNet/lib/core/function.py", line 52, in train
trainloader, optimizer, model, writer_dict)
File "/public/home/liw/Code/HRNet/lib/core/function.py", line 52, in train
for i_iter, batch in enumerate(trainloader, 0):
File "/public/home/liw/.conda/envs/HRNet/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 582, in next
for i_iter, batch in enumerate(trainloader, 0):
File "/public/home/liw/.conda/envs/HRNet/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 582, in next
return self._process_next_batch(batch)
File "/public/home/liw/.conda/envs/HRNet/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 608, in _process_next_batch
return self._process_next_batch(batch)
File "/public/home/liw/.conda/envs/HRNet/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 608, in _process_next_batch
raise batch.exc_type(batch.exc_msg)
AttributeError: Traceback (most recent call last):
File "/public/home/liw/.conda/envs/HRNet/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py", line 99, in _worker_loop
samples = collate_fn([dataset[i] for i in batch_indices])
File "/public/home/liw/.conda/envs/HRNet/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py", line 99, in
samples = collate_fn([dataset[i] for i in batch_indices])
File "/public/home/liw/Code/HRNet/lib/datasets/cityscapes.py", line 119, in getitem
label = self.convert_label(label)
File "/public/home/liw/Code/HRNet/lib/datasets/cityscapes.py", line 91, in convert_label
temp = label.copy()
AttributeError: 'NoneType' object has no attribute 'copy'

raise batch.exc_type(batch.exc_msg)

AttributeError: Traceback (most recent call last):
File "/public/home/liw/.conda/envs/HRNet/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py", line 99, in _worker_loop
samples = collate_fn([dataset[i] for i in batch_indices])
File "/public/home/liw/.conda/envs/HRNet/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py", line 99, in
samples = collate_fn([dataset[i] for i in batch_indices])
File "/public/home/liw/Code/HRNet/lib/datasets/cityscapes.py", line 119, in getitem
label = self.convert_label(label)
File "/public/home/liw/Code/HRNet/lib/datasets/cityscapes.py", line 91, in convert_label
temp = label.copy()
AttributeError: 'NoneType' object has no attribute 'copy'

Traceback (most recent call last):
File "/public/home/liw/.conda/envs/HRNet/lib/python3.6/runpy.py", line 193, in _run_module_as_main
"main", mod_spec)
File "/public/home/liw/.conda/envs/HRNet/lib/python3.6/runpy.py", line 85, in _run_code
exec(code, run_globals)
File "/public/home/liw/.conda/envs/HRNet/lib/python3.6/site-packages/torch/distributed/launch.py", line 235, in
main()
File "/public/home/liw/.conda/envs/HRNet/lib/python3.6/site-packages/torch/distributed/launch.py", line 231, in main
cmd=process.args)
subprocess.CalledProcessError: Command '['/public/home/liw/.conda/envs/HRNet/bin/python', '-u', 'tools/train.py', '--local_rank=0', '--cfg', 'experiments/cityscapes/seg_hrnet_w48_train_512x1024_sgd_lr1e-2_wd5e-4_bs_12_epoch484.yaml']' returned non-zero exit status 1.
End slurm job at Wed Nov 17 15:12:17 CST 2021