Correct way to upload saved model weights faster r-cnn

You could use forward hooks as described e.g. here, store the intermediate results in a dict or list for both models, and compare them using the same model input (and calling model.eval() to avoid random operations from e.g. dropout if these layers are used).

1 Like