Dear @xl-sr ,
I am training my custom dataset and I encountered the error above. I am not sure how to fix it even though I have looked at the code. Also, I only have 1 GPU, so I left --gpus=1 and --batch-gpu 64. The rest are the same as your instruction.
` File "/home/luu/.conda/envs/distilledDS_CL/lib/python3.11/site-packages/click/core.py", line 1055, in main
rv = self.invoke(ctx)
^^^^^^^^^^^^^^^^
File "/home/luu/.conda/envs/distilledDS_CL/lib/python3.11/site-packages/click/core.py", line 1404, in invoke
return ctx.invoke(self.callback, **ctx.params)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/luu/.conda/envs/distilledDS_CL/lib/python3.11/site-packages/click/core.py", line 760, in invoke
return __callback(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/luu/DistilledDataset_ContinualLearning/author_distilled_dataset/glad/stylegan_xl/train.py", line 321, in main
launch_training(c=c, desc=desc, outdir=opts.outdir, dry_run=opts.dry_run)
File "/home/luu/DistilledDataset_ContinualLearning/author_distilled_dataset/glad/stylegan_xl/train.py", line 104, in launch_training
subprocess_fn(rank=0, c=c, temp_dir=temp_dir)
File "/home/luu/DistilledDataset_ContinualLearning/author_distilled_dataset/glad/stylegan_xl/train.py", line 49, in subprocess_fn
training_loop.training_loop(rank=rank, **c)
File "/home/luu/DistilledDataset_ContinualLearning/author_distilled_dataset/glad/stylegan_xl/training/training_loop.py", line 208, in training_loop
'cur_nimg': torch.LongTensor([cur_nimg]),
UnboundLocalError: cannot access local variable 'cur_nimg' where it is not associated with a value
`
Dear @xl-sr , I am training my custom dataset and I encountered the error above. I am not sure how to fix it even though I have looked at the code. Also, I only have 1 GPU, so I left --gpus=1 and --batch-gpu 64. The rest are the same as your instruction.
` File "/home/luu/.conda/envs/distilledDS_CL/lib/python3.11/site-packages/click/core.py", line 1055, in main rv = self.invoke(ctx) ^^^^^^^^^^^^^^^^
File "/home/luu/.conda/envs/distilledDS_CL/lib/python3.11/site-packages/click/core.py", line 1404, in invoke return ctx.invoke(self.callback, **ctx.params) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/luu/.conda/envs/distilledDS_CL/lib/python3.11/site-packages/click/core.py", line 760, in invoke return __callback(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/luu/DistilledDataset_ContinualLearning/author_distilled_dataset/glad/stylegan_xl/train.py", line 321, in main launch_training(c=c, desc=desc, outdir=opts.outdir, dry_run=opts.dry_run)
File "/home/luu/DistilledDataset_ContinualLearning/author_distilled_dataset/glad/stylegan_xl/train.py", line 104, in launch_training subprocess_fn(rank=0, c=c, temp_dir=temp_dir)
File "/home/luu/DistilledDataset_ContinualLearning/author_distilled_dataset/glad/stylegan_xl/train.py", line 49, in subprocess_fn training_loop.training_loop(rank=rank, **c)
File "/home/luu/DistilledDataset_ContinualLearning/author_distilled_dataset/glad/stylegan_xl/training/training_loop.py", line 208, in training_loop 'cur_nimg': torch.LongTensor([cur_nimg]),
UnboundLocalError: cannot access local variable 'cur_nimg' where it is not associated with a value `