
代码技巧
XiaoPangJix1
这个作者很懒,什么都没留下…
展开
-
pytorch/tensorflow 直接给张量中的某个位置的值赋值,操作不可导。
参考:https://github.com/hadjisma/VideoAlignment/blob/master/d2tw/smoothDTW.py#L44。问题:给一个tensor A中[i,j],赋值p。解决方案:通过引入一个额外的mask实现。ps: 没debug, 看起来是对的。可能会导致值覆盖,操作不可导。原创 2023-08-29 12:01:11 · 1360 阅读 · 0 评论 -
Pytorch 记录每次循环输出的结果
需求:对每次循环生成的结果,stack,在循环结束后输出。原创 2023-08-03 09:39:27 · 195 阅读 · 0 评论 -
检查args中的键,有则自动赋值,如果没有则添加后赋值
var(args).keys()原创 2022-10-27 18:13:17 · 118 阅读 · 0 评论 -
将list作为参数,传入args
list作为args参数,传入函数。原创 2022-10-23 10:50:33 · 1035 阅读 · 0 评论 -
pytorch 模型中的bn层一键转化为同步bn(syncbn)
pytorch bn2syncbn原创 2022-07-28 14:31:49 · 1238 阅读 · 0 评论 -
pytorch 统计指定维度上的元素数目 torch.size numel
对给定的tensor,获取其指定几个维度上的张量元素数目。类型时,返回该tensor的元素数目;为统计元素数目的函数,当输入为。原创 2022-07-19 18:39:55 · 1568 阅读 · 0 评论 -
for + else
p ython中,for循环可以搭配else使用,以监测跳出循环的状态。例如:for i in range(N): if i < 5 : break else : print('i %d ' % i)如果在break中跳出循环,那么不进入else;如果没有在break中跳出循环(完整的遍历了所有i,自然地结束了for),那么就进入else。...原创 2022-04-29 09:57:42 · 190 阅读 · 0 评论 -
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 · 724 阅读 · 0 评论 -
python 移除字符串中首尾的指定字符串 strip
strip()用法:移除字符串中首尾的指定字符串,注意:出现在中间的字符串不会被移除;如果没有出现指定的字符串,也不会报错。'1d1ccd111'.strip('1')output: 'd1ccd''1d1ccd111'.strip('2')output: '1d1ccd111'...原创 2022-03-27 14:30:34 · 833 阅读 · 0 评论 -
约束多个变量相同
约束一堆数相同:len({a, b, c}) == 1利用{}unique的性质,只保留不同的数据。原创 2021-06-22 14:03:59 · 166 阅读 · 0 评论