- 博客(2)
- 收藏
- 关注
原创 BCEWithLogitsLoss样本不均衡的处理
最近在做deepfake检测任务(可以将其视为二分类问题,label为1和0),遇到了正负样本不均衡的问题,正样本数目是负样本的5倍,这样会导致FP率较高。尝试将正样本的loss权重增高,看BCEWithLogitsLoss的源码,Examples:: >>> target = torch.ones([10, 64], dtype=torch.float32...
2020-04-20 21:42:44 13306 5
原创 vs code中torch无法自动补全
问题一:最近接触pytorch(版本1.0),在vscode中发现写入import torch,下面会有红色波浪线,并且显示【pylint】Unable to import 'torch' [import-error],然后我试着import numpy、pandas等,也会出现类似的问题,同时引入这些库时无法自动补全。但是我在python交互界面是可以import torch和运行文件...
2019-03-04 18:57:14 10631 1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人