pytorch搭建网络，保存参数，恢复参数