I take my last comment back. Since a dropout layer is used in the network, the output after net.train() will not be exactly the same.
Setting the net to net.eval() before running gives the same output in both cases. So, at least network weights loading is correct.
Need to check if the issue is happening due to optimizer loading or data loader change after the runtime is reset. Thanks