-
Notifications
You must be signed in to change notification settings - Fork 37
Description
rzai@rzai00:~/prj/End-To-End-Generative-Dialogue/src$ CUDA_VISIBLE_DEVICES=0 th train.lua -data_file data/conv-train.hdf5 -val_data_file data/conv-val.hdf5 -save_file conv-model -gpuid 1
nil
Using CUDA on GPU 1...
Loading data...
Source vocab size: 10005, Target vocab size: 10005
Source max sent len: 7, Target max sent len: 7
Done loading data!
Building model with specs:
Layer type: lstm
Model type: red
Embedding size: 300
Hidden layer size: 300
Number of layers: 2
Using pre-trained word embeddings from data/word_vecs.hdf5
Number of parameters: 11902905
Beginning training...
Using ada_grad
/home/rzai/torch/install/bin/luajit: ...i/torch/install/share/lua/5.1/rnn/SequencerCriterion.lua:47: expecting target table
stack traceback:
[C]: in function 'assert'
...i/torch/install/share/lua/5.1/rnn/SequencerCriterion.lua:47: in function 'forward'
model_functions.lua:359: in function 'train_ind'
model_functions.lua:547: in function 'train_batch'
model_functions.lua:638: in function 'train'
train.lua:184: in main chunk
[C]: in function 'dofile'
...rzai/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:145: in main chunk
[C]: at 0x00406670
rzai@rzai00:~/prj/End-To-End-Generative-Dialogue/src$