今天有人问我,怎么保存模型的参数,下次直接调用。
结果,我发现,我连xshell也想不起来,也不会打开之前的jupyter notebook上面的.py文件了。
打开竟然都是这样子的 行了,放弃挣扎,直接请教我狮虎我狮虎,找出来我之前写的笔记:
行了,但是不记得网址了,也不记得密码了网址就是服务器地址啊~
行了,我又忘记密码了 行了,连我自己也没告诉 确认过眼神,是最傻的人重新来过:
在浏览器里面打开jupyter notebook:输入服务器地址:http://xx.xxx.xx.xxx:8888
输入密码就行了(真好,有记忆密码这个功能)
我在xshell里面的输入jupyter notebook就可以关掉了。以后直接在浏览器输入网址和密码就行了。
解决同学的问题:
这里,就是在保存参数,我的模型叫做model1,我用的是pytorch,输入语句torch.save(模型名字.state_dict(),'保存为的文件的名字’)如果保存整个模型的话,就不用加state_dict()
最后:
代码不能停啊~哈,又补充问道,如何读取这个参数啊:
就直接读取路径里面的这个文件就好了
网友评论