以下是我的代码
require 'dpnn'
require 'cunn'
local deviceNumber = tonumber(os.getenv("CUDA_CARD_NR"))
cutorch.setDevice(deviceNumber)
local module = nn.Sequential():cuda()
module:add(nn.Linear(2,1):cuda())
module:add(nn.Sigmoid():cuda())
criterion = nn.BCECriterion():cuda() -- Binary Cross Entorpy Criteria
local targets = torch.CudaTensor(10):random(0,1)
local inputs = torch.CudaTensor(10,2):uniform(-1,1)
function trainEpoch(module,criterion,inputs,targets)
for i=1,inputs:size(1) do
local idx = math.random(1,inputs:size(1))
local input, target = inputs[idx], targets:narrow(1,idx,1)
-- forward
local output= module:forward(input)
local loss= criterion:forward(output,target)
-- backward
local gradOutput = criterion:backward(output,target)
module:zeroGradParameters()
local gradInput = module:backward(input,gradOutput)
--update
module:updateGradParameters(0.9) -- momentum
module:updateParameters(0.1) -- W = W -0.1*dL/dW
end
end
for i=1,100 do
trainEpoch(module,criterion,inputs,targets)
end
我使用以下命令
在上面运行CUDA_CARD_NR=1 luajit feedforwad.lua
它出现以下错误
luajit: feedforwad.lua:13: attempt to call method 'random' (a nil value)
stack traceback:
feedforwad.lua:13: in main chunk
[C]: at 0x004064f0
我知道行中有一些错误
local targets = torch.CudaTensor(10):random(0,1)
但我无法理解。
答案 0 :(得分:0)
luajit:feedforwad.lua:13:尝试调用方法'随机'(一个零 值)
不是“一些错误”,你应该没有问题找出错误,因为错误信息告诉你到底出了什么问题。
您尝试调用名为random
的方法,该方法恰好是nil
值。
这意味着没有该名称的功能,因此您无法调用它。
根据参考文档(您应该在此处检查过),该函数实际上名为rand