Correct way to compare one model with different architectural implementation

I have to compare two different implementations of a network architecture with corresponding parameters. I know the number of parameters of the models is one way to make sure the implementations are similar. But, is there any other/ correct way to do this?
Thank you.

You could additionally try to print the graphs of both models (assuming you want to compare the forward/backward pass) and check if the computation graphs are also equal.

1 Like