Summing the model parameters and the parameters stored in the state_dict might yield a different result, since opt_level='O2' uses FP16 parameters inside the model, while the state_dict will contain upcasted FP32 parameters.
That being said, we recommend to use the native mixed precision implementation by installing the nightly binaries or build from master. The documentation can be found here 155.