PIRender icon indicating copy to clipboard operation
PIRender copied to clipboard

AttributeError: 'Config' object has no attribute 'device'

Open c1a1o1 opened this issue 3 years ago • 1 comments

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'

c1a1o1 avatar Jan 23 '22 12:01 c1a1o1

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

aminesoulaymani avatar Mar 28 '22 22:03 aminesoulaymani