OrgLion-ML
/
openface
mirror of https://github.com/cmusatyalab/openface.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Training: Handle peoplePerBatch > nClasses edge case when nDonkey != -1.

This commit is contained in:
Brandon Amos 2015-11-07 14:35:49 -05:00
parent ae348c22f9
commit 30f754176c
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
training/main.lua
View File

@ -28,12 +28,11 @@ paths.dofile('train.lua')
paths.dofile('test.lua')
paths.dofile('util.lua')
if opt.peoplePerBatch > #trainLoader.classes then
print('\n\nWarning: opt.peoplePerBatch > number of classes.')
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: ', #trainLoader.classes)
print('Setting opt.peoplePerBatch to the number of classes.\n\n')
opt.peoplePerBatch = #trainLoader.classes
print(' + number of classes: ', nClasses)
os.exit(-1)
end
epoch = opt.epochNumber