I am trying to train the kitti model with the exact configuration as yours, but I am getting an error. I will add the Traceback below, can you please help me what is wrong. TIA
`Starting initialization
Downloading: "https://download.pytorch.org/models/resnet18-5c106cde.pth" to /root/.cache/torch/checkpoints/resnet18-5c106cde.pth
100%|█████████████████████████████████████████████████████████████████████████████████| 46827520/46827520 [00:00<00:00, 94342063.16it/s]
Loading training dataset metadata:
Can use 57874 images from the kitti (kitti_split) train set for depth training
Can use 2975 images from the cityscapes train set for segmentation training
Loading validation dataset metadata:
Can use 1159 images from the kitti (kitti_split) validation set for depth validation
Can use 500 images from the cityscapes validation set for segmentation validation
Summary:
Using device: cuda (GeForce GTX 1080 Ti)
Epoch 0:
Traceback (most recent call last):
File "train.py", line 372, in
trainer.train()
File "train.py", line 340, in train
self._run_epoch()
File "train.py", line 228, in _run_epoch
for batch_idx, batch in enumerate(self.train_loaders):
File "/cmudi001-sgd-1/SGDepth/loaders/init.py", line 25, in iter
for batch_idx, group in zip(length_iter, infinite_iters):
File "/cmudi001-sgd-1/SGDepth/loaders/init.py", line 30, in _make_infinite
for batch in loader:
File "/opt/conda/envs/torch_110/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 582, in next
return self._process_next_batch(batch)
File "/opt/conda/envs/torch_110/lib/python3.7/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 "/opt/conda/envs/torch_110/lib/python3.7/site-packages/torch/utils/data/_utils/worker.py", line 99, in _worker_loop
samples = collate_fn([dataset[i] for i in batch_indices])
File "/opt/conda/envs/torch_110/lib/python3.7/site-packages/torch/utils/data/_utils/worker.py", line 99, in
samples = collate_fn([dataset[i] for i in batch_indices])
File "/opt/conda/envs/torch_110/lib/python3.7/site-packages/torch/utils/data/dataset.py", line 85, in getitem
return self.datasets[dataset_idx][sample_idx]
File "/cmudi001-sgd-1/SGDepth/dataloader/pt_data_loader/basedataset.py", line 217, in getitem
sample = self.add_const_dataset_items(sample)
File "/cmudi001-sgd-1/SGDepth/dataloader/pt_data_loader/specialdatasets.py", line 47, in add_const_dataset_items
native_im_shape = sample[native_key].shape
AttributeError: 'NoneType' object has no attribute 'shape'`
I am trying to train the kitti model with the exact configuration as yours, but I am getting an error. I will add the Traceback below, can you please help me what is wrong. TIA
`Starting initialization Downloading: "https://download.pytorch.org/models/resnet18-5c106cde.pth" to /root/.cache/torch/checkpoints/resnet18-5c106cde.pth 100%|█████████████████████████████████████████████████████████████████████████████████| 46827520/46827520 [00:00<00:00, 94342063.16it/s] Loading training dataset metadata: