GPU not being utilised

Hi @Pixel, low GPU utilisation might be because of small model or data size.