Skip to content

RuntimeError:cannot unsqueezze empty tensor #5

@Yealdor

Description

@Yealdor

in training process,I got error below:
Traceback (most recent call last):
File "/media/liye/volume0/Ubuntu/PythonProject/easy-fpn.pytorch/train.py", line 150, in
main()
File "/media/liye/volume0/Ubuntu/PythonProject/easy-fpn.pytorch/train.py", line 148, in main
_train(dataset_name, backbone_name, path_to_data_dir, path_to_checkpoints_dir, path_to_resuming_checkpoint)
File "/media/liye/volume0/Ubuntu/PythonProject/easy-fpn.pytorch/train.py", line 52, in train
for batch_index, (
, image_batch, _, bboxes_batch, labels_batch) in enumerate(dataloader):
File "/home/liye/anaconda3/envs/easy-fpn/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 336, in next
return self._process_next_batch(batch)
File "/home/liye/anaconda3/envs/easy-fpn/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 357, in _process_next_batch
raise batch.exc_type(batch.exc_msg)
RuntimeError: Traceback (most recent call last):
File "/home/liye/anaconda3/envs/easy-fpn/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 106, in _worker_loop
samples = collate_fn([dataset[i] for i in batch_indices])
File "/home/liye/anaconda3/envs/easy-fpn/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 187, in default_collate
return [default_collate(samples) for samples in transposed]
File "/home/liye/anaconda3/envs/easy-fpn/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 187, in
return [default_collate(samples) for samples in transposed]
File "/home/liye/anaconda3/envs/easy-fpn/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 164, in default_collate
return torch.stack(batch, 0, out=out)
RuntimeError: cannot unsqueeze empty tensor

Process finished with exit code 1

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions