(base) wheeltec-client@ubuntu:~/Desktop/ggcnn$ source activate py36
(py36) wheeltec-client@ubuntu:~/Desktop/ggcnn$ python3 train_ggcnn.py 
INFO:root:Loading Dataset...
>> train dataset: 86
>> train_val dataset: 8
>> test dataset: 9
INFO:root:Loading Network...
INFO:root:optimizer Done
----------------------------------------------------------------
        Layer (type)               Output Shape         Param #
================================================================
            Conv2d-1         [-1, 16, 360, 360]           1,952
              ReLU-2         [-1, 16, 360, 360]               0
            Conv2d-3         [-1, 16, 360, 360]           6,416
              ReLU-4         [-1, 16, 360, 360]               0
         MaxPool2d-5         [-1, 16, 180, 180]               0
            Conv2d-6         [-1, 16, 180, 180]           6,416
              ReLU-7         [-1, 16, 180, 180]               0
            Conv2d-8         [-1, 16, 180, 180]           6,416
              ReLU-9         [-1, 16, 180, 180]               0
        MaxPool2d-10           [-1, 16, 90, 90]               0
           Conv2d-11           [-1, 32, 90, 90]          12,832
             ReLU-12           [-1, 32, 90, 90]               0
           Conv2d-13           [-1, 32, 90, 90]          25,632
             ReLU-14           [-1, 32, 90, 90]               0
UpsamplingBilinear2d-15         [-1, 32, 180, 180]               0
           Conv2d-16         [-1, 16, 180, 180]           4,624
             ReLU-17         [-1, 16, 180, 180]               0
UpsamplingBilinear2d-18         [-1, 16, 360, 360]               0
           Conv2d-19         [-1, 16, 360, 360]           2,320
             ReLU-20         [-1, 16, 360, 360]               0
           Conv2d-21          [-1, 1, 360, 360]              17
           Conv2d-22          [-1, 1, 360, 360]              17
           Conv2d-23          [-1, 1, 360, 360]              17
           Conv2d-24          [-1, 1, 360, 360]              17
================================================================
Total params: 66,676
Trainable params: 66,676
Non-trainable params: 0
----------------------------------------------------------------
Input size (MB): 0.49
Forward/backward pass size (MB): 159.19
Params size (MB): 0.25
Estimated Total Size (MB): 159.94
----------------------------------------------------------------
INFO:root:Beginning Epoch 00, lr=0.001
[ WARN:0@0.765] global loadsave.cpp:244 findDecoder imread_('test/pcd0150d.tiff'): can't open/read file: check file path/integrity
[ WARN:0@0.772] global loadsave.cpp:244 findDecoder imread_('test/pcd0185d.tiff'): can't open/read file: check file path/integrity
[ WARN:0@0.782] global loadsave.cpp:244 findDecoder imread_('test/pcd0113d.tiff'): can't open/read file: check file path/integrity
Traceback (most recent call last):
  File "train_ggcnn.py", line 323, in <module>
    run()
  File "train_ggcnn.py", line 273, in run
    train_results = train(epoch, net, device, train_data, optimizer)
  File "train_ggcnn.py", line 142, in train
    for x, y in train_data:
  File "/home/wheeltec-client/anaconda3/envs/py36/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 521, in __next__
    data = self._next_data()
  File "/home/wheeltec-client/anaconda3/envs/py36/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 1203, in _next_data
    return self._process_data(data)
  File "/home/wheeltec-client/anaconda3/envs/py36/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 1229, in _process_data
    data.reraise()
  File "/home/wheeltec-client/anaconda3/envs/py36/lib/python3.6/site-packages/torch/_utils.py", line 434, in reraise
    raise exception
AttributeError: Caught AttributeError in DataLoader worker process 0.
Original Traceback (most recent call last):
  File "/home/wheeltec-client/anaconda3/envs/py36/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py", line 287, in _worker_loop
    data = fetcher.fetch(index)
  File "/home/wheeltec-client/anaconda3/envs/py36/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 49, in fetch
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/home/wheeltec-client/anaconda3/envs/py36/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 49, in <listcomp>
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/home/wheeltec-client/Desktop/ggcnn/utils/data/grasp_data.py", line 68, in __getitem__
    image.rescale(scale)
  File "/home/wheeltec-client/Desktop/ggcnn/utils/data/structure/img.py", line 165, in rescale
    self.img = mmcv.imrescale(self.img, scale, interpolation=interpolation)
  File "/home/wheeltec-client/Desktop/ggcnn/utils/dataset_processing/mmcv.py", line 130, in imrescale
    h, w = img.shape[:2]
AttributeError: 'NoneType' object has no attribute 'shape'

(py36) wheeltec-client@ubuntu:~/Desktop/ggcnn$