Faster RCNN - Number of epochs impact inference time

431 Views Asked by At

I have trained two Faster RCNN models with the exact same parameters with the exception of:

  1. Model 1: 150 epochs
  2. Model 2: 5 epochs

Once the models have been trained, I load them in and then I make one prediction on the same image for each model.

Inference prediction time:

  1. Model 1: 70 sec
  2. Model 2: 5 sec

To be clear, the time above is AFTER the model has been trained not during training.

Question: How can the inference time increase with the number of epochs?

Libraries:

  • tensorvision
  • torch
0

There are 0 best solutions below