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
Looks like --gpu -1 flag doesn't affect for program.
When I use this command: python src/run.py -c re.jpg -s 4.jpg --gpu -1
I receive this output: loading content image completed loading style image completed loading neural network model completed Traceback (most recent call last): File "src/run.py", line 51, in <module> style_transfer_runner.run(args) File "/home/z/cns/src/style_transfer_runner.py", line 73, in run out_image = model.fit(content_image, style_image, args.iter, on_epoch_done) File "/home/z/cns/src/neural_style.py", line 34, in fit with cuda.Device(device_id): AttributeError: 'module' object has no attribute 'Device'
Same output with command without --gpu flag at all
The text was updated successfully, but these errors were encountered:
jmic00
changed the title
Cpu only support?
Cpu mode support?
Jul 1, 2017
Looks like --gpu -1 flag doesn't affect for program.
When I use this command:
python src/run.py -c re.jpg -s 4.jpg --gpu -1
I receive this output:
loading content image completed
loading style image completed
loading neural network model completed
Traceback (most recent call last):
File "src/run.py", line 51, in <module>
style_transfer_runner.run(args)
File "/home/z/cns/src/style_transfer_runner.py", line 73, in run
out_image = model.fit(content_image, style_image, args.iter, on_epoch_done)
File "/home/z/cns/src/neural_style.py", line 34, in fit
with cuda.Device(device_id):
AttributeError: 'module' object has no attribute 'Device'
Same output with command without --gpu flag at all
The text was updated successfully, but these errors were encountered: