摘要: 方法1: 先加载两个pth: weights1 = torch.load("A.pth") weights2 = torch.load("A.pth") 查看keys: weights1.keys(),可能有state_dict, optimizer等, 通过字符串格式化,str(weights1[ 阅读全文
posted @ 2024-01-31 16:43 嶙羽 阅读(145) 评论(0) 推荐(0) 编辑
摘要: 习惯用notebook,但不知道咋回事突然开启服务的时候报错了 jupyter notebook --no-browser --port=1111 谷歌了下,有人提出了一个解决方案: https://github.com/jupyterhub/jupyterhub/issues/4418 试了下,真 阅读全文
posted @ 2023-11-08 14:29 嶙羽 阅读(8) 评论(0) 推荐(0) 编辑
摘要: 查看vim /etc/ssh/sshd_config 发现密码验证(PasswordAuthentication)关闭了 重启ssh服务systemctl restart sshd.service 选择自己账户下的ssh ID,然后就可以在本地使用ssh密码登陆了 阅读全文
posted @ 2023-10-29 16:47 嶙羽 阅读(32) 评论(0) 推荐(0) 编辑
摘要: 因为很难直接把服务器上的大量图片下到本地,需要在服务器上查看, python -m http.server 8888 启动web服务器后,在本地浏览器上打开地址: 服务器地址:8888,就可以在浏览器上进入图片所在目录进行查看 阅读全文
posted @ 2023-04-11 14:54 嶙羽 阅读(54) 评论(0) 推荐(0) 编辑
摘要: 论文里需要用到t-SNE对特征进行可视化,原理还没细看,但总体思路比较清晰,具体实现网上也有现成的代码。 对输出层的前一层特征提取出来,比如某个batch,维度是(b, 300),对应的target也提取出来,维度(b, k),注意这里需要把one-hot转化为类别形式, 注意,需要把整个测试集都过 阅读全文
posted @ 2022-03-30 11:25 嶙羽 阅读(649) 评论(0) 推荐(0) 编辑
摘要: 首先在本地(MAC)生成公钥和私钥: ssh-keygen -t rsa 发现在本地/Users/Yourname/.ssh下生成了id_rsa和id_rsa.pub两个文件,分别是私钥和公钥 然后我们需要把公钥放到服务器账户下的对应位置: ssh-copy-id -i /Yourname/.ssh 阅读全文
posted @ 2022-02-17 14:32 嶙羽 阅读(530) 评论(0) 推荐(0) 编辑
摘要: 之前一直不知道Jupyter Notebook里怎样使用argparse传参数,直到今天终于google了一下,发现只需要替换一步。。 # opt = parser.parse_args() opt, unknown = parser.parse_known_args() 参考: https://f 阅读全文
posted @ 2022-02-09 21:41 嶙羽 阅读(347) 评论(0) 推荐(0) 编辑
摘要: https://www.jianshu.com/p/1afd25e7459d 阅读全文
posted @ 2022-02-09 21:30 嶙羽 阅读(61) 评论(0) 推荐(0) 编辑
摘要: 最近遇到了一个问题,模型是用DistributedDataParallel一机多卡分布式训练的,然后作为一个jupyter notebook重度用户,我想用它来加载这个模型,搞点预测的例子可视化看看。 但是这会碰到一个问题,我们都知道通常加载预训练模型的方法是: pretrained_dict = 阅读全文
posted @ 2021-07-02 10:56 嶙羽 阅读(1039) 评论(0) 推荐(0) 编辑
摘要: Pytorch使用这个方法,就可以在已有模型的基础上重新加一些结构,比如已经加载了某个预训练模型,但仍需要增加一些模块,此时已经不能在一开始构建模型的时候init进去模块,于是可以在load完模型后,用model.add_module,把所需要的模块增添进去,使用方法是model.add_modul 阅读全文
posted @ 2021-03-12 09:41 嶙羽 阅读(1096) 评论(0) 推荐(0) 编辑