55 lines
1.1 KiB
Lua
Executable File
55 lines
1.1 KiB
Lua
Executable File
#!/usr/bin/env th
|
|
|
|
require 'torch'
|
|
require 'optim'
|
|
|
|
require 'paths'
|
|
|
|
require 'xlua'
|
|
|
|
local opts = paths.dofile('opts.lua')
|
|
|
|
opt = opts.parse(arg)
|
|
print(opt)
|
|
|
|
if opt.cuda then
|
|
require 'cutorch'
|
|
cutorch.setDevice(1)
|
|
end
|
|
|
|
if opt.cudnn then
|
|
print("\nThere is a known issue with cudnn training! See:")
|
|
print(" https://github.com/cmusatyalab/openface/issues/127\n\n")
|
|
end
|
|
|
|
os.execute('mkdir -p ' .. opt.save)
|
|
torch.save(paths.concat(opt.save, 'opts.t7'), opt, 'ascii')
|
|
print('Saving everything to: ' .. opt.save)
|
|
|
|
torch.setdefaulttensortype('torch.FloatTensor')
|
|
|
|
torch.manualSeed(opt.manualSeed)
|
|
|
|
paths.dofile('data.lua')
|
|
paths.dofile('util.lua')
|
|
paths.dofile('model.lua')
|
|
paths.dofile('train.lua')
|
|
paths.dofile('test.lua')
|
|
|
|
if opt.peoplePerBatch > nClasses then
|
|
print('\n\nError: opt.peoplePerBatch > number of classes. Please decrease this value.')
|
|
print(' + opt.peoplePerBatch: ', opt.peoplePerBatch)
|
|
print(' + number of classes: ', nClasses)
|
|
os.exit(-1)
|
|
end
|
|
|
|
epoch = opt.epochNumber
|
|
|
|
for _=1,opt.nEpochs do
|
|
train()
|
|
if opt.testing then
|
|
test()
|
|
end
|
|
epoch = epoch + 1
|
|
end
|