机器学习
能向左或向右
这个作者很懒,什么都没留下…
展开
-
爱斯维尔模板错误总结——坑爹货!
总结爱斯维尔模板里编译不通过的一些小错误,通常是doc太多看不懂造成的。原创 2023-01-18 17:59:04 · 1871 阅读 · 4 评论 -
pytorch——计算网络模型参数个数
parameters = filter(lambda p: p.requires_grad, model.parameters()) # for parameter in parameters: # print(np.prod(parameter.size()) / 1_000_000) parameters = sum([np.prod(p.size()) for p in parameters]) / 1_000_000 print('Trainable ...原创 2021-07-10 09:24:06 · 559 阅读 · 0 评论 -
pytorch自定义损失函数
相信在看到这篇文章之前,你已经看到了各位博主都再说pytorch自定已损失函数很简单,主要通过以下三个手段:原创 2021-05-31 15:15:27 · 992 阅读 · 0 评论 -
只有一层lstm,Out of memory
事情是这样的,我的网络结构是这样的:只有一层lstm+avgpooling+linear,且linear只有一个32*4的隐藏层。网络层很小,所以肯定不是网络参数的事儿,那么问题就出现在样本上。训练数据是有标签样本每批16篇文章,无标签样本5000+,测试集是(有标签+无标签)*0.2。input并没有被torch.cuda.empty_cache()清除 。因此,在使用完网络输出结果后,直接del,这样就不会超出内存了。...原创 2021-05-26 21:42:45 · 232 阅读 · 0 评论 -
pytorch训练中出现的错误汇集
错误一: File "D:\anaconda3\envs\PyTorch\lib\site-packages\torch\nn\modules\rnn.py", line 559, in forward max_batch_size = input.size(0) if self.batch_first else input.size(1)IndexError: Dimension out of range (expected to be in range of [-1, 0], but .原创 2020-11-15 17:56:30 · 5368 阅读 · 6 评论