Commit Graph

18 Commits

Author SHA1 Message Date
melgor 4fad635aa4 Fix testing 2016-07-13 19:28:11 +02:00
melgor 0995e86b7e Added Multi-GPU support #106 2016-07-04 22:23:37 +02:00
Brandon Amos 92684ab771 Training: Add cuda device option. 2016-06-14 15:55:46 -04:00
Brandon Amos fea2fb41d8 Resolve nan issues for #127 (hopefully) 2016-06-13 13:50:38 -04:00
Brandon Amos c6f0962b6d Training: Add -save option. Default to a timestamp instead of numbers. 2016-06-13 13:50:38 -04:00
Brandon Amos bced2bb445 Disable cudnn by default for #127. Add warning. 2016-06-07 18:20:46 -04:00
Brandon Amos 1b3b48f588 Load util before model for optimizeNet
\cc @melgor
2016-04-03 02:48:50 -04:00
Brandon Amos 04d23481fb Fix typo. 2016-03-06 21:15:10 -05:00
Brandon Amos bbff0a01dd Fix tests for training the DNN. 2016-03-06 19:45:37 -05:00
Brandon Amos f5766412d2 NN Training: Add LFW validation for #100. 2016-03-04 18:30:39 -05:00
Brandon Amos e987b15b22 Move old test code to attic for #99. 2016-02-27 17:50:11 -05:00
Brandon Amos 15cc5fef00 Training: Add CPU execution for tests. 2016-01-11 16:17:36 -05:00
Brandon Amos 2769411d56 Add @melgor's training improvements for #55 2016-01-06 17:02:30 -05:00
Brandon Amos 57118ac28f Fix luacheck warnings. 2015-12-27 08:41:49 -05:00
Brandon Amos e90694fcba nn4: Add LRN and remove 5x5 convolutions for #55.
Reduces the number of parameters from 7472144 to 6959088.
2015-12-20 22:29:55 -05:00
Brandon Amos 30f754176c Training: Handle peoplePerBatch > nClasses edge case when nDonkey != -1. 2015-11-07 14:35:49 -05:00
Brandon Amos 9bef1e8a60 Train: Detect and correct edge case when peoplePerBatch > number of classes.
Thanks @venturamartin90
2015-11-06 12:10:25 -05:00
Brandon Amos d8790408d1 Initial commit of training code from development repo. 2015-09-23 19:49:45 -04:00