- 博客(64)
- 收藏
- 关注
RSS订阅原创 pytorch 模型去掉最后一层 或 最后几层
pytorch的模型去掉末端的n层，以n=1为例：new_model = torch.nn.Sequential( *( list(model.children())[:-1] ) )ps: 模型如何去掉任意位置的某一层？好像不支持？可能去掉之后，图就改变了，问题比较复杂。但是可以通过在load statedict时选择性的加载有需要的参数层，来间接实现这一目的。...
2022-04-18 09:01:59
5169
原创 shell scipt: for + call function
#! /bin/bashexport CUDA_VISIBLE_DEVICES="0"test_cmd (){ python proto.py \ --test_model True \ --checkpoint $i \ --dataset somethingotam \ --model ResNet50_pretrained \ --L2norm False \ --test_episode 5000 \}for i in $1/epoch* do test_c
2022-04-15 10:15:50
727
原创 install opencv from tsinghua mirror
cv2 package install, (through tsinghua mirror)pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple
2022-04-04 09:04:13
3460
原创 pytorch L2 norm
进行L2norm，i.e., 计算范数，除以范数。torch.nn.functional.normalize(feature, p=2, dim=1)```
2022-03-27 16:49:15
2995
原创 python 移除字符串中首尾的指定字符串 strip
strip()用法：移除字符串中首尾的指定字符串，注意：出现在中间的字符串不会被移除；如果没有出现指定的字符串，也不会报错。'1d1ccd111'.strip('1')output: 'd1ccd''1d1ccd111'.strip('2')output: '1d1ccd111'...
2022-03-27 14:30:34
839
原创 pytorch 手动设置参数变量 并转到cuda上
通过nn.Parameter设置参数（参与后向，可更新，随模型保存）self.temprt_Q = torch.nn.Parameter(torch.tensor(1.).cuda(), requires_grad=True)注意⚠️：转到cuda的设置要在将tensor转成parameter之前，即：torch.tensor(1.).cuda()，而不是torch.nn.Parameter(torch.tensor(1.), requires_grad=True).cuda(),该操作会导致para
2022-01-11 20:13:37
4367
原创 ‘SSLError(SSLCertVerificationError(1, ‘[SSL: CERTIFICATE_VERIFY_FAILED] certificate verify failed:
pip 安装包包错，网站不信任，需要验证。报错：WARNING: Retrying (Retry(total=0, connect=None, read=None, redirect=None, status=None)) after connection broken by 'SSLError(SSLCertVerificationError(1, '[SSL: CERTIFICATE_VERIFY_FAILED] certificate verify failed: certificate is n
2021-12-28 10:06:22
4854
原创 借助vim统计文本中的字符串数目，行首字符串查询
普通命令：:%s/pattern//gnpattern为要查询的字符串技巧：当需要统计的字符串处于行首，且是其他字符串的子集时，可以考虑每行行首加一个特定字符，使得待查询的字符串变得特定。举例：4321abc111111187654321abcfdsafcdsf假定4321abc是待查询的字符串且需要位于行首，而下一行中也有该字符串，但不位于行首。因此可以通过对每行插入一串特殊字符，使得位于行首的4321abc变得特定。注意执行下述命令前需要cp一个备份。例如利用命令:%s/^/__vv
2021-11-03 21:29:48
3672
原创 arxiv上传overleaf的latex源码, arxiv上传踩坑与注意事项
在arxiv上传的整体流程可参考知乎链接本文只针对arxiv对latex源码的一些特殊要求进行调整清除所有error由于overleaf的编译功能很强大，所以即便编译过程有error，也可以正常生成pdf。但arxiv在线编译过程严格要求不能有错。所以第一步是要解决工程中固有的报错，在overleaf中编译不报错即可。文章目录清除所有error参考文献`bib` to `bbl`注释掉tex中对bib文件的设置bbl 文件写入图片：eps to pdf一些其他的tipsarxiv账号注册arxiv
2021-10-27 21:50:55
12859
原创 pytorch 程序设置随机种子
设置随机种子可以控制每次程序运行时的随机性保持一致，使得结果可复现。def seed_torch(seed=1029): random.seed(seed) os.environ['PYTHONHASHSEED'] = str(seed) np.random.seed(seed) torch.manual_seed(seed) torch.cuda.manual_seed(seed) torch.cuda.manual_seed_all(seed) # if you are using mu
2021-10-04 22:10:16
386
原创 vi下批量删除字符串 vim
利用vi编辑器可批量删除文本文件中指定的字符串（注意只删除该字符串，而不是删除它出现过的行）。如，删除ad123，vi编辑器在命令模式下输入：:%s/ad123注意，如果待删除的关键字中含有/，那么命令中需要更换分隔符/(可换成#），比如，删除/home/abc123::%s#/home/abc123...
2021-04-27 14:57:50
8483
原创 杀进程：按照应用名杀进程；杀死某个gpu上的所有进程
按照应用名称杀死一批进程，以python应用为例：ps -ef|grep python|grep -v grep|cut -c 9-15|xargs kill -9name: application name杀死某个指定gpu上的所有进程 ， 以gpu2为例：fuser -v /dev/nvidia2 |awk '{for(i=1;i<=NF;i++)print "kill -9 " $i;}' | sh......
2020-12-08 12:03:17
5808
8
原创 打印当前日期&时间
打印当前日期&时间import datetimetime_stamp = datetime.datetime.now()print "time: " + time_stamp.strftime('%Y.%m.%d-%H:%M:%S') #strftime可以自定义时间的输出格式#输出为time_stamp 2017.02.19-14:03:20
2020-12-03 16:17:54
387
原创 PYTORCH torch.einsum 函数
以爱因斯坦求和的形式任意定义想要进行的矩阵乘法的操作，可以内部指定输出转置，功能多样，形式灵活。#theta_phi: nxtxg#g: nxcxgoutput = torch.einsum('ntg, ncg->nct', theta_phi, g)#output: nxcxt
2020-09-10 15:15:38
2154
原创 移动硬盘挂载报错unable to mount unknown file system type exfat
硬盘挂载报错:unable to mount xxxunknown file system type exfat解决方法：sudo apt-get install exfat-fuse
2020-08-07 21:44:12
1021
原创 linux更换内核后重启找不到显卡驱动
问题：待机很久的机器重新启动后，屏幕分辨率极低，watch nvidia-smi 命令报错。怀疑是找不到驱动了。解决方案：参考：([https://blog.csdn.net/qq_33656324/article/details/107031291] [https://blog.csdn.net/qq_33656324/article/details/107031291]产生问题的原因是之前更换了内核，导致重启后新的内核无法指向现在的驱动。step1安装dmks模块，帮助新内核指向现有驱动
2020-08-07 12:01:14
1579
1
原创 Mac下压缩文件夹后，在windows解压后出现一个额外的__MACOSX文件夹。
Mac下压缩文件夹后，在windows解压后出现一个额外的__MACOSX文件夹：终端中执行命令，不压缩 __MACOSC文件夹unzip shell.zip shell -x __MACOSX/*
2020-05-15 21:41:52
12258
原创 linux下 teamviewer 安装过程报错errors were encountered while processing: teamviewer
安装命令：sudo dpkg -i teamviewer_amd64.deb报错：(最后一行）errors were encountered while processing:teamviewer解决方法：sudo dpkg --add-architecture i386sudo apt-get updatesudo apt-get -f install重新执行安装命令：...
2020-04-27 15:46:41
1634
原创 caffe2 operator
Append将两个tensor在第一个维度上拼接，将第一个维度视作batch的维度，要求：in-place output ，输出的blob和输入的第一个blob一致。input1: a1xbxcinput2: a2xbxcoutput:(input1) (a1+a2)xbxcBatchGather...
2019-10-14 22:25:51
390
原创 non-local caffe2 编译
non-local caffe2 环境设置欢迎使用Markdown编辑器一些必要的编译知识ffmpeg + opencv准备non-local中需要的包功能快捷键合理的创建标题，有助于目录的生成如何改变文本的样式插入链接与图片如何插入一段漂亮的代码片生成一个适合你的列表创建一个表格设定内容居中、居左、居右SmartyPants创建一个自定义列表如何创建一个注脚注释也是必不可少的KaTeX数学公式新...
2019-08-02 21:48:04
1054
原创 Kinetics 数据库处理
Kinetics 数据库处理下载mp4 --> jpg比较3D conv与non-local的设置差异本文仅为自己研究过程的一个随手记录，部分细节可能没有严格考证kinetics数据库是当前较大规模的行为识别数据库，400类，training set 246k video, validation set 20k. 由于此前没有接触过视频数据库，而这个库又非常大，处理起来比较麻烦，记录一下...
2019-01-19 22:36:58
3328
38
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人