产生式模型和判别式模型(Generative model vs. Discriminative model),概率图模型(Graphical Models),朴素贝叶斯分类器( Naive Bayes Classifier),隐马尔可夫模型(Hidden Markov Model,HMM),最大熵模型(Maximum Entropy Model,MEM),最大熵马尔可夫模型(MEMM),条件随机场(conditional random fields,CRF)
跑模型时出现RuntimeError: CUDA out of memory.错误 查阅了许多相关内容,原因是:GPU显存内存不够
简单总结一下解决方法:
将batch_size改小。
取torch变量标量值时使用item()属性。
可以在测试阶段添加如下代码:
with torch.no_grad(): # 停止自动反向计算梯度
参考:
https://wisdomai.xyz/tool/pytorch/archives/2301
https://ptorch.com/news/160.html
作者:菜叶儿掉啦