You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I got this bug when using the --single-gpu flag.
I juste added :
opt.device = torch.cuda.current_device()
I got another bug about local_rank, I just added:
opt.distributed = False
F:\ProgramData\Anaconda3\python.exe F:/work/mingxingshipin/test/PIRender-main/intuitive_control.py --config ./config/face_demo.yaml --name face --no_resume --output_dir ./vox_result/face_intuitive --input_name ./demo_images
Make folder result\face
2022-01-23 20:13:22.076873: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_100.dll
Traceback (most recent call last):
File "F:/work/mingxingshipin/test/PIRender-main/intuitive_control.py", line 87, in
= get_model_optimizer_and_scheduler(opt)
File "F:\work\mingxingshipin\test\PIRender-main\util\trainer.py", line 46, in get_model_optimizer_and_scheduler
net_G = network(**opt.gen.param).to(opt.device)
AttributeError: 'Config' object has no attribute 'device'
The text was updated successfully, but these errors were encountered: