Hi,
I would say that this is expected that a relatively small mnist model won’t use much memory both on CPU and GPU.
CPU utilization is expected to be quite low if you run the main code on GPU. It will only load data from the disk and launch work on the GPU.
GPU utilization will depend on the model again, but with a batch_size of 10, you shouldn’t expect to see a big usage. Also keep in mind that this value is an average so it might be 100% for short burst then waiting on the CPU to load more data from the disk.