torch save state dict

Solutions on MaxInterview for torch save state dict by the best coders in the world