Open hellozgm opened 5 years ago
@hellozgm Same problem! Have you found the solution?
The multi GPUs is not supported currently. If you want to accelerate the training process, please use the hyperparameter batchSize while training. But the result shows that batchSize=1 can perform better.
Hi, thank for your awesome work. I want to modify the code to fit the multi GPU process, and I modify your main code below:
But, I got error:
I have no idea about this problem, do you have some suggestions? Thank you!