/home/sdnu/anaconda3/envs/DDIM/bin/python /media/sdnu/f9cc3556-f530-42b2-95df-64c823288321/home/sdnu/SXY/Diffusion/latent-dehazing/train.py
export CUDA_VISIBLE_DEVICES=0
Disabled distributed training.
Path already exists. Rename it to [/media/sdnu/f9cc3556-f530-42b2-95df-64c823288321/home/sdnu/experiments/latent-dehazing/latent-reffusion-dehazing_archived_231219-163532]
cosine schedule
(600, 400, 3) (600, 400, 3)
(400, 600, 3) (400, 600, 3)
(400, 600, 3) (400, 600, 3)
(400, 600, 3) (400, 600, 3)
(400, 600, 3) (400, 600, 3)
(600, 400, 3) (600, 400, 3)
(400, 600, 3) (400, 600, 3)
(400, 600, 3) (400, 600, 3)
(600, 400, 3) (600, 400, 3)
(400, 600, 3) (400, 600, 3)
(400, 600, 3) (400, 600, 3)
(400, 600, 3) (400, 600, 3)
(600, 400, 3) (600, 400, 3)
(400, 600, 3) (400, 600, 3)
(600, 400, 3) (600, 400, 3)
(400, 600, 3) (400, 600, 3)
(400, 600, 3) (400, 600, 3)
(400, 600, 3) (400, 600, 3)
(600, 400, 3) (600, 400, 3)
(400, 600, 3) (400, 600, 3)
(600, 400, 3) (600, 400, 3)
(600, 400, 3) (600, 400, 3)
(600, 400, 3) (600, 400, 3)
(400, 600, 3) (400, 600, 3)
(600, 400, 3) (600, 400, 3)
(600, 400, 3) (600, 400, 3)
(400, 600, 3) (400, 600, 3)
(400, 600, 3) (400, 600, 3)
(600, 400, 3) (600, 400, 3)
(600, 400, 3) (600, 400, 3)
(400, 600, 3) (400, 600, 3)
(600, 400, 3) (600, 400, 3)
(400, 600, 3) (400, 600, 3)
(400, 600, 3) (400, 600, 3)
(600, 400, 3) (600, 400, 3)
(600, 400, 3) (600, 400, 3)
(600, 400, 3) (600, 400, 3)
(600, 400, 3) (600, 400, 3)
(400, 600, 3) (400, 600, 3)
(600, 400, 3) (600, 400, 3)
(400, 600, 3) (400, 600, 3)
(400, 600, 3) (400, 600, 3)
(400, 600, 3) (400, 600, 3)
(400, 600, 3) (400, 600, 3)
(600, 400, 3) (600, 400, 3)
(600, 400, 3) (600, 400, 3)
(600, 400, 3) (600, 400, 3)
(600, 400, 3) (600, 400, 3)
(600, 400, 3) (600, 400, 3)
(400, 600, 3) (400, 600, 3)
(400, 600, 3) (400, 600, 3)
(600, 400, 3) (600, 400, 3)
(400, 600, 3) (400, 600, 3)
(400, 600, 3) (400, 600, 3)
(600, 400, 3) (600, 400, 3)
(600, 400, 3) (600, 400, 3)
Traceback (most recent call last):
File "/media/sdnu/f9cc3556-f530-42b2-95df-64c823288321/home/sdnu/SXY/Diffusion/latent-dehazing/train.py", line 326, in <module>
main()
File "/media/sdnu/f9cc3556-f530-42b2-95df-64c823288321/home/sdnu/SXY/Diffusion/latent-dehazing/train.py", line 229, in main
for _, train_data in enumerate(train_loader):
File "/home/sdnu/anaconda3/envs/DDIM/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 517, in __next__
data = self._next_data()
File "/home/sdnu/anaconda3/envs/DDIM/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 1199, in _next_data
return self._process_data(data)
File "/home/sdnu/anaconda3/envs/DDIM/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 1225, in _process_data
data.reraise()
File "/home/sdnu/anaconda3/envs/DDIM/lib/python3.7/site-packages/torch/_utils.py", line 429, in reraise
raise self.exc_type(msg)
RuntimeError: Caught RuntimeError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/home/sdnu/anaconda3/envs/DDIM/lib/python3.7/site-packages/torch/utils/data/_utils/worker.py", line 202, in _worker_loop
data = fetcher.fetch(index)
File "/home/sdnu/anaconda3/envs/DDIM/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 47, in fetch
return self.collate_fn(data)
File "/home/sdnu/anaconda3/envs/DDIM/lib/python3.7/site-packages/torch/utils/data/_utils/collate.py", line 73, in default_collate
return {key: default_collate([d[key] for d in batch]) for key in elem}
File "/home/sdnu/anaconda3/envs/DDIM/lib/python3.7/site-packages/torch/utils/data/_utils/collate.py", line 73, in <dictcomp>
return {key: default_collate([d[key] for d in batch]) for key in elem}
File "/home/sdnu/anaconda3/envs/DDIM/lib/python3.7/site-packages/torch/utils/data/_utils/collate.py", line 55, in default_collate
return torch.stack(batch, 0, out=out)
RuntimeError: stack expects each tensor to be equal size, but got [3, 600, 400] at entry 0 and [3, 400, 600] at entry 7
Process finished with exit code 1