随笔分类 -  pytorch

CTC+pytorch编译配置warp-CTC
摘要:CTCCTC可以生成一个损失函数,用于在序列数据上进行监督式学习,不需要对齐输入数据及标签,经常连接在一个RNN网络的末端,训练端到端的语音和文本识别系统。CTC论文地址: http://www.cs.toronto.edu/%7Egraves/icml_2006.p... 阅读全文
posted @ 2018-05-04 21:22 未雨愁眸 阅读(1122) 评论(0) 推荐(0)
pytorch在CPU和GPU上加载模型
摘要:pytorch允许把在GPU上训练的模型加载到CPU上,也允许把在CPU上训练的模型加载到GPU上。CPU->CPU,GPU->GPUtorch.load('gen_500000.pkl')GPU->CPUtorch.load('gen_500000.pkl', ma... 阅读全文
posted @ 2018-03-10 09:00 未雨愁眸 阅读(1503) 评论(0) 推荐(0)