随笔分类 - pytorch
CTC+pytorch编译配置warp-CTC
摘要:CTCCTC可以生成一个损失函数,用于在序列数据上进行监督式学习,不需要对齐输入数据及标签,经常连接在一个RNN网络的末端,训练端到端的语音和文本识别系统。CTC论文地址: http://www.cs.toronto.edu/%7Egraves/icml_2006.p...
阅读全文
pytorch在CPU和GPU上加载模型
摘要:pytorch允许把在GPU上训练的模型加载到CPU上,也允许把在CPU上训练的模型加载到GPU上。CPU->CPU,GPU->GPUtorch.load('gen_500000.pkl')GPU->CPUtorch.load('gen_500000.pkl', ma...
阅读全文
浙公网安备 33010602011771号