Run training on a pytorch models using 1 a100rome gpu. Record the images per second and add it to the plot for regression testing/observation.

tests: cuda, pytorch