When running the interpreter's training, the error below appears for me. Does anyone know what is going on?
Traceback (most recent call last):
File "train_interpreter.py", line 579, in
main(opts)
File "train_interpreter.py", line 447, in main
all_feature_maps_train_all, all_mask_train_all, num_data = prepare_data(args, palette)
File "train_interpreter.py", line 403, in prepare_data
img, feature_maps = latent_to_image(g_all, upsamplers, latent_input.unsqueeze(0), dim=args['dim'][1],
File ".../datasetGAN_release/datasetGAN/../utils/utils.py", line 76, in latent_to_image
img_list, affine_layers = g_all.module.g_synthesis(style_latents)
File ".../.conda/envs/dg/lib/python3.8/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(*input, kwargs)
File ".../datasetGAN_release/datasetGAN/../models/stylegan1.py", line 558, in forward
x, x2 = m(dlatents_in[:, 2 i:2 i + 2])
File ".../.conda/envs/dg/lib/python3.8/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(*input, *kwargs)
File ".../datasetGAN_release/datasetGAN/../models/stylegan1.py", line 370, in forward
x = self.conv(x)
File ".../.conda/envs/dg/lib/python3.8/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(input, kwargs)
File ".../mspacheco/datasetGAN_release/datasetGAN/../models/stylegan1.py", line 125, in forward
return F.conv2d(x, self.weight * self.w_mul, bias, padding=self.kernel_size // 2)
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
When running the interpreter's training, the error below appears for me. Does anyone know what is going on?
Traceback (most recent call last): File "train_interpreter.py", line 579, in
main(opts)
File "train_interpreter.py", line 447, in main
all_feature_maps_train_all, all_mask_train_all, num_data = prepare_data(args, palette)
File "train_interpreter.py", line 403, in prepare_data
img, feature_maps = latent_to_image(g_all, upsamplers, latent_input.unsqueeze(0), dim=args['dim'][1],
File ".../datasetGAN_release/datasetGAN/../utils/utils.py", line 76, in latent_to_image
img_list, affine_layers = g_all.module.g_synthesis(style_latents)
File ".../.conda/envs/dg/lib/python3.8/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(*input, kwargs)
File ".../datasetGAN_release/datasetGAN/../models/stylegan1.py", line 558, in forward
x, x2 = m(dlatents_in[:, 2 i:2 i + 2])
File ".../.conda/envs/dg/lib/python3.8/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(*input, *kwargs)
File ".../datasetGAN_release/datasetGAN/../models/stylegan1.py", line 370, in forward
x = self.conv(x)
File ".../.conda/envs/dg/lib/python3.8/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(input, kwargs)
File ".../mspacheco/datasetGAN_release/datasetGAN/../models/stylegan1.py", line 125, in forward
return F.conv2d(x, self.weight * self.w_mul, bias, padding=self.kernel_size // 2)
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED