pytorch
ystsaan
这个作者很懒,什么都没留下…
展开
-
AttributeError: module ‘onnxruntime‘ has no attribute ‘InferenceSession‘
因为py文件名字被我命名成了onnxruntime.py与python库onnxruntime重名了,修改onnxruntime.py为onnxruntime_test.py再执行就好了。。。原创 2021-04-28 19:20:06 · 1552 阅读 · 0 评论 -
OHEM,Focal loss,GHM loss二分类pytorch代码实现(减轻难易样本不均衡问题)
https://mp.weixin.qq.com/s/iOAICJege2b0pCVxPkvNiA综述:解决目标检测中的样本不均衡问题该综述主要介绍了OHEM,Focal loss,GHM loss;由于我这的二分类数据集不存在正负样本不均衡的问题,所以着重看了处理难易样本不均衡(正常情况下,容易的样本较多,困难的样本较少);由于我只是分类问题,所以写了各种分类的loss,且网络的最后一层为softmax,所以网络输出的pred是softmax层前的logits经过softmax后的结果,普通的交叉熵损原创 2020-10-16 16:49:44 · 5227 阅读 · 3 评论 -
CrossEntropyLoss,BCEWithLogitsLoss,BCEWithLogitsLoss细节(给不同的样本赋予不同的权重;给不同类别的样本赋予不同的权重)
torch.nn.BCELoss,torch.nn.BCEWithLogitsLoss中的weight参数和torch.nn.CrossEntropyLoss的weight参数意义不一样,torch.nn.CrossEntropyLoss,torch.nn.BCEWithLogitsLoss的weight是每个class的权重,torch.nn.BCELoss中的weight是每个样本的权重;torch.nn.BCEWithLogitsLoss对于输入都要做一个sigmoid操作再做average(-ylo原创 2020-10-15 11:03:30 · 2489 阅读 · 0 评论 -
pyslowfast multigrid训练自己的数据集(没有把网络模型给抠出来再按照自己的训练,而是完全按照作者的代码命令训练)
安装:具体参考https://github.com/facebookresearch/SlowFast/blob/master/INSTALL.md注意点:1.使用conda install pytorch torchvision cudatoolkit=9.2在自己服务器的虚拟环境下安装pytorch时需要使得cudatoolkit与驱动版本(使用nvidia-smi查看)相匹配,不然torch.cuda.is_available()返回false使得安装的pytorch无法使用gpu加速;版本配对原创 2020-07-21 19:03:23 · 2798 阅读 · 6 评论 -
torch.cuda.is_available()返回false
按照https://pytorch.org/使用下列命令安装pytorch时cudatoolkit版本写10.0,不要写10.1或10.2(1.3 1.4和10.0配对时torch.cuda.is_available()正常)conda install pytorch torchvision cudatoolkit=10.0原创 2020-07-16 09:04:20 · 382 阅读 · 0 评论 -
cyclegan(pix2pix)训练测试
github:https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix在网站中有以下一段话,原来以为使用自己的图片集需要改pytorch-CycleGAN-and-pix2pix/data/template_dataset.py,其实只要把数据集存放的文件夹名字改成和example dataset一样就可以了Custom Model an...原创 2020-01-22 11:12:03 · 2114 阅读 · 5 评论 -
Default process group has not been initialized, please make sure to call init_process_group.
1原创 2019-11-14 15:34:57 · 43885 阅读 · 9 评论