网站首页 Pytorch

• 

  Pytorch训练模型得到输出后计算F1-Score 和AUC的操作

  1、计算F1-Score对于二分类来说，假设batchsize大小为64的话，那么模型一个batch的输出应该是torch.size([64,2])，所以首先做的是得到这个二维矩阵的每一行的最大索引值，然后添加到一个列表中，同时把标签也...

  2021-07-12 longtao100 python 519 ℃ 0 评论
• 

  Pytorch训练网络过程中loss突然变为0的解决方案

  问题// loss 突然变成0python train.py -b=8INFO: Using device cpuINFO: Network:&...

  2021-07-12 longtao100 python 450 ℃ 0 评论
• 

  解决Numpy与Pytorch彼此转换时的坑

  前言  最近使用Numpy包与Pytorch写神经网络时，经常需要两者彼此转换，故用此笔记记录码代码时踩（菜）过的坑，网上有人说：Pytorch又被称为GPU版的Numpy，二者的许多功能都有良好的一一对应。但在使用时还是得多多...

  2021-07-12 longtao100 python 727 ℃ 0 评论