您的位置:首页 > 编程语言 > Lua

Self summary: Using Lua in torch (keep updating)

2016-03-29 21:03 381 查看
1/ When you are creating a target of training samples, it must be a tensor.

2/ If you want to use cuda to compute, the data that input the net:forward/backward and criterion forward/backward must be cuda tensor. And net and criterion can use net:cuda() directly to become the net for cuda. However, data like tensor must be transformed
by data = data:cuda(). Keep in mind.  
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  lua cuda