Can you provide a small example of how the outputs differ so we can get a better idea of what the issue is? It may also be worthwhile to inspect the intermediate outputs of the model (and the “augmentations” depending on what is in opt.augment) to see where the differences first appear.