pytorch
ucas_fhx
我要一步一步往上爬!
展开
-
BCEWithLogitsLoss样本不均衡的处理
最近在做deepfake检测任务(可以将其视为二分类问题,label为1和0),遇到了正负样本不均衡的问题,正样本数目是负样本的5倍,这样会导致FP率较高。尝试将正样本的loss权重增高,看BCEWithLogitsLoss的源码,Examples:: >>> target = torch.ones([10, 64], dtype=torch.float32...原创 2020-04-20 21:42:44 · 13550 阅读 · 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 · 10698 阅读 · 1 评论