xh-liu / CC-FPSE

Code for NeurIPS 2019 paper "Learning to Predict Layout-to-image Conditional Convolutions for Semantic Image Synthesis"
128 stars 15 forks source link

Cant reproduce results, self.instance_path is empty. Please help me #12

Open Kitty-sunray opened 2 years ago

Kitty-sunray commented 2 years ago

Hello! I try to run train with the command train.py --name name --mpdist --netG condconv --dist_url tcp://0.0.0.0:8000 --num_servers 1 --netD fpse --lambda_feat 20 --dataset_mode custom --dataroot=path --batchSize 1 --niter 100 --niter_decay 100 --use_vae --ngpus_per_node 1 --label_dir=path --image_dir=path --checkpoints_dir=path --no_vgg_loss

And gettin the error I could not trace back: Use GPU: 0 for training dataset [CustomDataset] of size 14544 was created Network [CondConvGenerator] was created. Total number of parameters: 135.6 million. To see the architecture, do print(network). Network [FPSEDiscriminator] was created. Total number of parameters: 5.2 million. To see the architecture, do print(network). Network [ConvEncoder] was created. Total number of parameters: 10.5 million. To see the architecture, do print(network). create web directory path... Traceback (most recent call last): File "train.py", line 97, in main() File "train.py", line 29, in main mp.spawn(main_worker, nprocs=ngpus_per_node, args=(ngpus_per_node, idx_server, opt)) File "/opt/conda/lib/python3.7/site-packages/torch/multiprocessing/spawn.py", line 199, in spawn return start_processes(fn, args, nprocs, join, daemon, start_method='spawn') File "/opt/conda/lib/python3.7/site-packages/torch/multiprocessing/spawn.py", line 157, in start_processes while not context.join(): File "/opt/conda/lib/python3.7/site-packages/torch/multiprocessing/spawn.py", line 118, in join raise Exception(msg) Exception:

-- Process 0 terminated with the following error: Traceback (most recent call last): File "/opt/conda/lib/python3.7/site-packages/torch/multiprocessing/spawn.py", line 19, in _wrap fn(i, *args) File "/content/CC-FPSE/train.py", line 58, in main_worker for i, data_i in enumerate(dataloader, start=iter_counter.epoch_iter): File "/opt/conda/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 435, in next data = self._next_data() File "/opt/conda/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 475, in _next_data data = self._dataset_fetcher.fetch(index) # may raise StopIteration File "/opt/conda/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 "/opt/conda/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 44, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/content/CC-FPSE/data/pix2pix_dataset.py", line 79, in getitem instance_path = self.instance_paths[index]

How to avid that? Also, there is no vgg19.pth can be found anywhere. What file supposed to be used?