UNet Training is not executed on GPU

[peterjc123 (Pu Jiachen)]

Switch to the Compute node and you’ll see the usage of your CUDA cores.