this is trained on over 600 images of GPUs. training has ended