test pytorch in golang Prepare local dev env Saving and loading models https://pytorch.org/tutorials/beginner/saving_loading_models.html#what-is-a-state-dict