search

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
Other
3.09k stars 682 forks source link

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

Open liwei1101 opened 2 years ago

liwei1101 commented 2 years ago

=> 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