Hi, thanks for sharing this project.
I have encounted such error when I try to use another dataset(nuScene) to train the model.
Traceback (most recent call last):
File "train.py", line 58, in <module>
start(ap.parse_args())
File "train.py", line 41, in start
trainer.fit(model, dataset)
File "/root/anaconda3/envs/comogan/lib/python3.7/site-packages/pytorch_lightning/trainer/trainer.py", line 510, in fit
results = self.accelerator_backend.train()
File "/root/anaconda3/envs/comogan/lib/python3.7/site-packages/pytorch_lightning/accelerators/accelerator.py", line 57, in train
return self.train_or_test()
File "/root/anaconda3/envs/comogan/lib/python3.7/site-packages/pytorch_lightning/accelerators/accelerator.py", line 74, in train_or_test
results = self.trainer.train()
File "/root/anaconda3/envs/comogan/lib/python3.7/site-packages/pytorch_lightning/trainer/trainer.py", line 561, in train
self.train_loop.run_training_epoch()
File "/root/anaconda3/envs/comogan/lib/python3.7/site-packages/pytorch_lightning/trainer/training_loop.py", line 542, in run_training_epoch
for batch_idx, (batch, is_last_batch) in train_dataloader:
File "/root/anaconda3/envs/comogan/lib/python3.7/site-packages/pytorch_lightning/profiler/profilers.py", line 85, in profile_iterable
value = next(iterator)
File "/root/anaconda3/envs/comogan/lib/python3.7/site-packages/pytorch_lightning/trainer/connectors/data_connector.py", line 46, in _with_is_last
last = next(it)
File "/opt/data/private/test/ideals/CoMoGAN/data/__init__.py", line 82, in __iter__
for i, data in enumerate(self.dataloader):
File "/root/anaconda3/envs/comogan/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 435, in __next__
data = self._next_data()
File "/root/anaconda3/envs/comogan/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 1085, in _next_data
return self._process_data(data)
File "/root/anaconda3/envs/comogan/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 1111, in _process_data
data.reraise()
File "/root/anaconda3/envs/comogan/lib/python3.7/site-packages/torch/_utils.py", line 428, in reraise
raise self.exc_type(msg)
NameError: Caught NameError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/root/anaconda3/envs/comogan/lib/python3.7/site-packages/torch/utils/data/_utils/worker.py", line 198, in _worker_loop
data = fetcher.fetch(index)
File "/root/anaconda3/envs/comogan/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/root/anaconda3/envs/comogan/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 44, in <listcomp>
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/opt/data/private/test/ideals/CoMoGAN/data/day2timelapse_dataset.py", line 91, in __getitem__
A = self.transform_A(A_img)
File "/root/anaconda3/envs/comogan/lib/python3.7/site-packages/torchvision/transforms/transforms.py", line 67, in __call__
img = t(img)
File "/root/anaconda3/envs/comogan/lib/python3.7/site-packages/torchvision/transforms/transforms.py", line 393, in __call__
return self.lambd(img)
File "/opt/data/private/test/ideals/CoMoGAN/data/base_dataset.py", line 83, in <lambda>
transform_list.append(transforms.Lambda(lambda img: __make_power_2(img, base=4, method=method)))
File "/opt/data/private/test/ideals/CoMoGAN/data/base_dataset.py", line 107, in __make_power_2
__print_size_warning(ow, oh, w, h)
File "/opt/data/private/test/ideals/CoMoGAN/data/base_dataset.py", line 138, in __print_size_warning
logger.warning("The image size needs to be a multiple of 4. "
NameError: name 'logger' is not defined
Hi, thanks for sharing this project. I have encounted such error when I try to use another dataset(nuScene) to train the model.