Closed JJun-Guo closed 2 years ago
hello do you train and eval on cpu?
GPU
where can change the gpu id !!!?
I have only one gpu so did not set this... You need to add this code.
hello do you train and eval on cpu?