Created by: huawang123
I added a few lines of code to specify the GPU, hoping it would be useful, and changed the name because data.py and the data folder would conflict
Created by: huawang123
I added a few lines of code to specify the GPU, hoping it would be useful, and changed the name because data.py and the data folder would conflict