Torch7学习笔记[2] ---神经网络的主体框架
来源:互联网 发布:servlet修改表单数据 编辑:程序博客网 时间:2024/05/18 12:43
参考资料:
https://github.com/soumith/cvpr2015/blob/master/Deep%20Learning%20with%20Torch.ipynb
将整个框架分为以下几个模块:
1、options设置
2、train、test预处理以及读取
3、net结构以及criterion建立
4、train设置
5、test设置
6、保存中间结果以及断点开始(待完善)
y以上每个功能模块单独由一个文件完成,整个结构分为7个文件
main.lua
opt.lua
dataloder.lua
model.lua
train.lua
test.lua
checkpont.lua(待完善)
require 'torch'require 'nn'require 'optim'local DataLoder = require 'dataloder' --load the dataloder.lualocal opts = require 'opt'local Model = require 'model'local Test = require 'test'local checkpoints = require 'checkpoint'local Trainer = require 'train'torch.setdefaulttensortype = ('torch.FloatTensor') --torch.setnumthreads(1)torch.manualSeed(opt.manualSeed)cutorch.manualSeedAll(opt.manualSeed)local opt = opts.parse(arg) --load the optionslocal trainset,testset = DataLoder.creat(opt) --load the datasetlocal model,criterion = Model.setup(opt) --load the model,criterionif(opt.type == 'cuda') then --turn on gpu:model-criterion-data-label model = model:cuda() criterion = criterion:cuda() trainset.data = trainset.data:cuda() trainset.label = trainset.label:cuda() testset.data = testset.data:cuda() testset.label = testset.label:cuda()endfunction trainset:size() --prepare for training return self.data:size(1) endlocal trainer = Trainer(model,criterion,opt)bestModel = falsefor epoch = 1,opt.max_epoch do local current_error = trainer:train(epoch,trainset) --save the current station --checkpoints.save(epoch, model, trainer.optimState, bestModel, opt)endlocal correct_rate = Test.run(opt,testset,model)print(correct_rate)
运行程序时,直接在文件所在目录终端执行:th main.lua 即可运行程序。若需改变options,例如gpu运行:th main.lua –type cuda
0 0
- Torch7学习笔记[2] ---神经网络的主体框架
- Torch7框架及XNOR-Net学习笔记(2)--iTorch的安装
- Torch7学习(四)——学习神经网络包的用法(2)
- Torch7 学习笔记[1]
- Torch7下搭建卷积神经网络框架
- HTML5 学习笔记2-新增的主体结构元素
- Torch7框架学习资料整理
- 安装torch7 深度学习框架
- Torch7学习(三)——学习神经网络包的用法(1)
- 深度学习框架Torch7解析-- Tensor篇
- Torch7 学习笔记[3] --option setting
- imooc网的主体框架
- Torch7学习(六)——学习神经网络包的用法(4)——利用optim进行训练
- HTML学习笔记之--HTML主体的常用设置
- Torch7 学习
- torch7学习笔记1——itorch_notebook与lua编译的安装
- 【SQL Server学习笔记】Windows主体、SQL Server主体、数据库主体
- libev 学习笔记之主体事件循环
- 利用ImageView和SeekBar动态缩放和旋转图片的Demo
- 析构函数调用顺序
- 主题: kindeditor加入ckplayer视频播放器的插件使用讲解
- Ubuntu15.1编译Android5.1详解
- iOS/OS X 内存管理(二):借助工具解决内存问题
- Torch7学习笔记[2] ---神经网络的主体框架
- 对Android 回调机制的一些理解
- OpenCV 阈值化
- 关于实现抽取公共DAO
- [完]Python函数 extend()和append()的区别
- Android清除本地数据缓存代码
- android自动提示功能——autoCompleteTextView(一)
- Kindeditor(版本号4.0.5)编辑器添加上传flv视频功能
- iOS开发:代码片段