19-08-19 20:31:34.055 - INFO: Model [SRRaGANModel] is created.
19-08-19 20:31:34.055 - INFO: Start training from epoch: 0, iter: 0
export CUDA_VISIBLE_DEVICES=0
Path already exists. Rename it to [..\experiments\Cool_Model_archived_190819-203129]
torch.Size([3, 32, 32])
torch.Size([3, 32, 32])
torch.Size([3, 32, 32])
torch.Size([3, 32, 32])
torch.Size([3, 32, 32])
torch.Size([3, 32, 32])
torch.Size([3, 32, 32])
torch.Size([3, 32, 32])
torch.Size([3, 32, 32])
torch.Size([3, 32, 32])
torch.Size([3, 32, 32])
torch.Size([3, 32, 32])
torch.Size([3, 32, 32])
torch.Size([3, 32, 32])
torch.Size([3, 32, 32])
torch.Size([3, 32, 32])
torch.Size([3, 0, 0])
torch.Size([3, 0, 0])
torch.Size([3, 0, 0])
torch.Size([3, 116, 0])
torch.Size([3, 0, 0])
torch.Size([3, 0, 128])
torch.Size([3, 0, 0])
torch.Size([3, 0, 96])
torch.Size([3, 128, 0])
torch.Size([3, 0, 0])
torch.Size([3, 0, 128])
torch.Size([3, 0, 0])
torch.Size([3, 0, 24])
torch.Size([3, 0, 128])
torch.Size([3, 0, 0])
torch.Size([3, 0, 48])
Traceback (most recent call last):
  File "train.py", line 176, in <module>
    main()
  File "train.py", line 100, in main
    for _, train_data in enumerate(train_loader):
  File "C:\Python37\lib\site-packages\torch\utils\data\dataloader.py", line 346, in __next__
    data = self.dataset_fetcher.fetch(index)  # may raise StopIteration
  File "C:\Python37\lib\site-packages\torch\utils\data\_utils\fetch.py", line 47, in fetch
    return self.collate_fn(data)
  File "C:\Python37\lib\site-packages\torch\utils\data\_utils\collate.py", line 77, in default_collate
    return {key: default_collate([d[key] for d in batch]) for key in elem}
  File "C:\Python37\lib\site-packages\torch\utils\data\_utils\collate.py", line 77, in <dictcomp>
    return {key: default_collate([d[key] for d in batch]) for key in elem}
  File "C:\Python37\lib\site-packages\torch\utils\data\_utils\collate.py", line 58, in default_collate
    return torch.stack(batch, 0, out=out)
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 0 and 116 in dimension 2 at C:\w\1\s\windows\pytorch\aten\src\TH/generic/THTensor.cpp:689