How to increase GPU usage during training

Related to this: