furthermore when I run my code (on a small network that fits in ram) I get (relatively) low gpu usage:

image.png939×848 32.3 KB

What could be the cause of this? Shouldnt it be able to run at close to a 100%?