nerv.include('reader.lua')
nerv.include('timer.lua')
nerv.include('config.lua')
nerv.include(arg[1])

local global_conf = get_global_conf()
local timer = global_conf.timer

local data_path = 'examples/lmptb/PTBdata/'

local layers = get_layers(global_conf)
local connections = get_connections(global_conf)

local NN = nerv.NN(global_conf, layers, connections)

timer:tic('global')
local best_cv = 1e10
for i = 1, global_conf.max_iter do
    timer:tic('Epoch' .. i)
    local train_reader = nerv.Reader(data_path .. 'vocab', data_path .. 'ptb.train.txt.adds')
    local val_reader = nerv.Reader(data_path .. 'vocab', data_path .. 'ptb.valid.txt.adds')
    local train_ppl, val_ppl = NN:epoch(train_reader, val_reader)
    if val_ppl < best_cv then
        best_cv = val_ppl
    else
        global_conf.lrate = global_conf.lrate / 2.0
    end
    nerv.printf('Epoch %d: %f %f %f\n', i, global_conf.lrate, train_ppl, val_ppl)
    timer:toc('Epoch' .. i)
    timer:check('Epoch' .. i)
    io.flush()
end
timer:toc('global')
timer:check('global')
timer:check('network')
timer:check('gc')
timer:check('IO')
global_conf.cumat_type.print_profile()