EMA weights at inference

[fses91]

Hi,

in your sim_inference.py , you load the model checkpoint class which calls the on_load_checkpoint(self, checkpoint) callback and loads the EMA weights into your ExponentialMovingAverage object, but the load_ema_weights(self) function is not called anymore which should replace the model weights with the ema weights.

Am I correct, that your inference script is not using the EMA weights?

Best regards