Traceback (most recent call last):
File "E:\deeplearning\stylegan3\stylegan3-main\train.py", line 296, in
main()# pylint: disable=no-value-for-parameter
File "D:\anaconda\install\envs\stylegan3\lib\site-packages\click\core.py", line 1130, in call
return self.main(args, kwargs)
File "D:\anaconda\install\envs\stylegan3\lib\site-packages\click\core.py", line 1055, in main
rv = self.invoke(ctx)
File "D:\anaconda\install\envs\stylegan3\lib\site-packages\click\core.py", line 1404, in invoke
return ctx.invoke(self.callback, ctx.params)
File "D:\anaconda\install\envs\stylegan3\lib\site-packages\click\core.py", line 760, in invoke
return __callback(args, kwargs)
File "E:\deeplearning\stylegan3\stylegan3-main\train.py", line 281, in main
launch_training(c=c, desc=desc, outdir=opts.outdir, dry_run=opts.dry_run)
File "E:\deeplearning\stylegan3\stylegan3-main\train.py", line 96, in launch_training
subprocess_fn(rank=0, c=c, temp_dir=temp_dir)
File "E:\deeplearning\stylegan3\stylegan3-main\train.py", line 47, in subprocess_fn
training_loop.training_loop(rank=rank, c)
File "E:\deeplearning\stylegan3\stylegan3-main\training\training_loop.py", line 278, in training_loop
loss.accumulate_gradients(phase=phase.name, real_img=real_img, real_c=real_c, gen_z=gen_z, gen_c=gen_c, gain=phase.interval, cur_nimg=cur_nimg)
File "E:\deeplearning\stylegan3\stylegan3-main\training\loss.py", line 138, in accumulate_gradients
(loss_Dreal + loss_Dr1).mean().mul(gain).backward()
File "D:\anaconda\install\envs\stylegan3\lib\site-packages\torch_tensor.py", line 255, in backward
torch.autograd.backward(self, gradient, retain_graph, create_graph, inputs=inputs)
File "D:\anaconda\install\envs\stylegan3\lib\site-packages\torch\autograd__init__.py", line 147, in backward
Variable._execution_engine.run_backward(
File "D:\anaconda\install\envs\stylegan3\lib\site-packages\torch\autograd\function.py", line 87, in apply
return self._forward_cls.backward(self, *args) # type: ignore[attr-defined]
File "D:\anaconda\install\envs\stylegan3\lib\site-packages\torch\autograd\function.py", line 195, in backward
raise NotImplementedError("You must implement the backward function for custom"
NotImplementedError: You must implement the backward function for custom autograd.Function.
Process finished with exit code 1
Screenshots
If applicable, add screenshots to help explain your problem.
Desktop (please complete the following information):
OS: [e.g. Linux Ubuntu 20.04, Windows 10]
PyTorch version (e.g., pytorch 1.9.0)
CUDA toolkit version (e.g., CUDA 11.1)
NVIDIA driver version
GPU [e.g., Titan V, RTX 3090]
Additional context
I think there is a problem with backpropagation. I have tried many versions, but it is still the case.
Describe the bug Training for 5000 kimg...
Traceback (most recent call last): File "E:\deeplearning\stylegan3\stylegan3-main\train.py", line 296, in
main()# pylint: disable=no-value-for-parameter
File "D:\anaconda\install\envs\stylegan3\lib\site-packages\click\core.py", line 1130, in call
return self.main(args, kwargs)
File "D:\anaconda\install\envs\stylegan3\lib\site-packages\click\core.py", line 1055, in main
rv = self.invoke(ctx)
File "D:\anaconda\install\envs\stylegan3\lib\site-packages\click\core.py", line 1404, in invoke
return ctx.invoke(self.callback, ctx.params)
File "D:\anaconda\install\envs\stylegan3\lib\site-packages\click\core.py", line 760, in invoke
return __callback(args, kwargs)
File "E:\deeplearning\stylegan3\stylegan3-main\train.py", line 281, in main
launch_training(c=c, desc=desc, outdir=opts.outdir, dry_run=opts.dry_run)
File "E:\deeplearning\stylegan3\stylegan3-main\train.py", line 96, in launch_training
subprocess_fn(rank=0, c=c, temp_dir=temp_dir)
File "E:\deeplearning\stylegan3\stylegan3-main\train.py", line 47, in subprocess_fn
training_loop.training_loop(rank=rank, c)
File "E:\deeplearning\stylegan3\stylegan3-main\training\training_loop.py", line 278, in training_loop
loss.accumulate_gradients(phase=phase.name, real_img=real_img, real_c=real_c, gen_z=gen_z, gen_c=gen_c, gain=phase.interval, cur_nimg=cur_nimg)
File "E:\deeplearning\stylegan3\stylegan3-main\training\loss.py", line 138, in accumulate_gradients
(loss_Dreal + loss_Dr1).mean().mul(gain).backward()
File "D:\anaconda\install\envs\stylegan3\lib\site-packages\torch_tensor.py", line 255, in backward
torch.autograd.backward(self, gradient, retain_graph, create_graph, inputs=inputs)
File "D:\anaconda\install\envs\stylegan3\lib\site-packages\torch\autograd__init__.py", line 147, in backward
Variable._execution_engine.run_backward(
File "D:\anaconda\install\envs\stylegan3\lib\site-packages\torch\autograd\function.py", line 87, in apply
return self._forward_cls.backward(self, *args) # type: ignore[attr-defined]
File "D:\anaconda\install\envs\stylegan3\lib\site-packages\torch\autograd\function.py", line 195, in backward
raise NotImplementedError("You must implement the backward function for custom"
NotImplementedError: You must implement the backward function for custom autograd.Function.
Process finished with exit code 1
Screenshots If applicable, add screenshots to help explain your problem.
Desktop (please complete the following information):
Additional context I think there is a problem with backpropagation. I have tried many versions, but it is still the case.