style-based-gan-pytorch icon indicating copy to clipboard operation
style-based-gan-pytorch copied to clipboard

training error

Open Passenger12138 opened this issue 2 years ago • 1 comments

I downloaded the source code for stylegan and tried to run it on my machine. but the model cannot be successfully trained. This is my error message My machine is configured with 10 RTX 3090 GPUs, image My code running environment is python3.6, pytorch1.3.1.torchvision 0.4.2 image I tried to specify the gpu for training, but there is still an error message image

Passenger12138 avatar May 06 '22 08:05 Passenger12138

update pytorch version

ttytu2 avatar Mar 19 '23 14:03 ttytu2