编程
小刀丶
这个作者很懒,什么都没留下…
展开
-
[BUG 记录] torch.multinomial/ CUDA error: device-side assert triggered
[BUG 记录] torch.multinomial/ CUDA error: device-side assert triggered使用torch.multinomial时需要保证可取数量大于采样数,否则有些概率趋近于0的项永远不会被采集到。解决办法:1.设置replacement为True2.对概率做个限制(clamp),防止某些决策的概率趋近于0...原创 2020-09-16 18:59:21 · 362 阅读 · 0 评论 -
[BUG记录] pytorch 内存占用量与张量尺寸不对等
今天利用torch.save储存了一个size为[1, 2048, 20, 33]的torch.short类型的CPU张量在磁盘里,读取的时候写torch.load(......)[..., :1, :], 结果发现内存占用没有变为原来的1/20解决方案:切片后加.contiguous()...原创 2019-12-22 20:05:39 · 743 阅读 · 0 评论 -
Leet Code 力扣 - - 最短最优雅python解法带解析汇总
Leet Code 刷题笔记记录一下刷过的有意义和代表性的题动态规划Unique Paths题目可以转换为排列组合问题,解是C(min(m,n), m+n),从m+n个中选出m个下移或n个右移。用DP做很快,以后算 C(a, b) 都可以用算这题的DP法代替...原创 2019-04-19 05:42:20 · 4119 阅读 · 0 评论