Hey. The number of model weights and number of batches (of only 2-dimensional data samples) are too low to saturate all GPU memory and compute. You can try to increase the batch size to see the increase of GPU utilization.