2015-09-28 71 views
1

我是新來的火炬。 我試圖運行Torch.But單一分類實驗,我收到以下錯誤,當訓練開始時,火炬中的單分類器網絡

/手電筒/安裝/斌/ luajit:壞參數#2「?」 (超出範圍在/torch/pkg/torch/generic/Tensor.c:853)

棧回溯:

[C]: at 0x7f17b9dc029 
    [C]: in function '__index' 
    .../torch/install/share/lua/5.1/optim/ConfusionMatrix.lua:40: in function '_add' 
    .../torch/install/share/lua/5.1/optim/ConfusionMatrix.lua:102: in function 'batchAdd' 
    Main.lua:246: in function 'Train' 
    Main.lua:289: in main chunk 
    [C]: in function 'dofile' 
    .../torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:131: in main chunk 
    [C]: at 0x00406670 

是否有可能在火炬訓練單一分類網絡?
在此先感謝。

+1

嘿,在看不到它的情況下幫助你的腳本是相當困難的。也許你應該附上這個「單分類器實驗」的源代碼? –

回答

0

感謝您的回覆。 我的模型包含,

分類= nn.Sequential()

分類:添加(nn.Dropout(0.5))

分類:添加(nn.Linear(512512))

分類:添加(nn.BatchNormalization(512))

分類:添加(nn.ReLU(真))

分類:添加(nn.Dropout(0.5))

分類:加(nn.Linear(512,1))

分類:加(nn.LogSoftMax())

,我使用nn.CrossEntropyCriterion()在虧損網絡。

是否可以運行單個分類器實驗?

0

是的,你可以在一門課上進行訓練。

你得到的錯誤是指向混淆矩陣。 對於一類應該以下列方式: - 類

classes = {1} 

- 這個矩陣記錄跨類當前混亂

confusion = optim.ConfusionMatrix(classes) 

您的火車標籤或測試標籤不應該包含' 0'作爲標籤。