Argument parser does not recognise mps

[rociorey]

Your issue may already be reported!
Please search on the issue tracker before creating one.

Context

I am running

python neural_style/neural_style.py train --dataset "path" --style-image "image-path" --save-model-dir "path"
 --epochs 2 --cuda 1

• Pytorch version: 2.0.1 with CUDA 11.8
• Operating System and version: Windows 11 Pro

Your Environment

• Installed using source? [yes/no]:
• Are you planning to deploy it using docker container? [yes/no]: yes
• Is it a CPU or GPU environment?: GPU
• Which example are you using: fast-neural-style

Current Behavior

I get the error AttributeError: 'Namespace' object has no attribute 'mps'

I have tried to set the gpu id to 0 as referenced in issue number 1134, still not working.

[msaroufim]

Hmm I think this was a recent regression #1064

Could you try commenting out the mps specific bits from the code and see if that solves the problem?