Torch Save Model Vs State Dict