You would first have to isolate the bottleneck before a solution can be found.
E.g. if the data loading is the bottleneck, you would need to speed it up by e.g. using multiple workers, speeding up the transformations etc.



 twister9458:

It raises the % of utilisation of GPU from 1.7GB to 3.2GB


The utilization shows a percentage of the time the GPU was busy in a specific last time frame, not the memory usage.