Maybe your model structure is too simple for GPU. It means you need to cost more time on data transfer and the improvement brought by GPU is very limited. It is normal. Try to use complex models and more batch_size for this test.
Topic:
Machine Learning & AI
SubTopic:
General
Tags: