Training on kitti data is not working

[chetanmreddy]

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:
• Model name: sgdepth_chetan
• Logging directory: /cmudi001-sgd-1/SGDepth/Checkpoints/sgdepth_chetan_test/sgdepth_chetan
• 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'`