when I run train.py file it is running on CPU directly. can you show the command how to run train.py on GPU?
when I run train.py file it is running on CPU directly.
can you show the command how to run train.py on GPU?