-
Notifications
You must be signed in to change notification settings - Fork 73
/
main.lua
27 lines (24 loc) · 633 Bytes
/
main.lua
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
local Setup = require 'Setup'
local Master = require 'Master'
local AsyncMaster = require 'async/AsyncMaster'
local AsyncEvaluation = require 'async/AsyncEvaluation'
-- Parse options and perform setup
local setup = Setup(arg)
local opt = setup.opt
-- Start master experiment runner
if opt.async then
if opt.mode == 'train' then
local master = AsyncMaster(opt)
master:start()
elseif opt.mode == 'eval' then
local eval = AsyncEvaluation(opt)
eval:evaluate()
end
else
local master = Master(opt)
if opt.mode == 'train' then
master:train()
elseif opt.mode == 'eval' then
master:evaluate()
end
end