Skip to content

pytorch_infer

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

tests: cuda, pytorch, anaconda3_gpu