Commit Graph

31 Commits

Author SHA1 Message Date
melgor 0995e86b7e Added Multi-GPU support #106 2016-07-04 22:23:37 +02:00
Brandon Amos a1b3251f96 Actually fix nans from #127. Error if they appear. 2016-06-14 13:51:18 -04:00
Brandon Amos c04386c48c Training: Only convert to cuda in cuda mode. 2016-06-13 14:12:36 -04:00
Brandon Amos f4567e882e Fixes for travis. 2016-06-13 14:06:12 -04:00
Brandon Amos fea2fb41d8 Resolve nan issues for #127 (hopefully) 2016-06-13 13:50:38 -04:00
Brandon Amos 21f1bcc5c4 Temporarily remove optimize_net for #127. 2016-06-07 18:14:13 -04:00
Brandon Amos 130edcecc9 Training: Use ADAM instead of adadelta. 2016-06-03 17:48:38 -04:00
melgor bba911c802 Replace satitize by clearState() 2016-03-16 18:20:11 +01:00
melgor 7d21caf004 Merge branch 'cudnn_convert' of https://github.com/melgor/openface into cudnn_convert
Conflicts:
	training/model.lua
	training/train.lua
2016-03-16 16:15:05 +01:00
melgor f63acb637e Replace nn_to_cudnn by cudnn.convert #106 2016-03-16 15:38:06 +01:00
melgor e2d7513ff5 Replace nn_to_cudnn by cudnn.convert #106 2016-03-11 11:09:23 +01:00
Brandon Amos f5766412d2 NN Training: Add LFW validation for #100. 2016-03-04 18:30:39 -05:00
Brandon Amos 9b04bca65d train.lua: Pass on comments. 2016-01-16 23:50:54 -05:00
Brandon Amos 51b01c200c Train: Fix model saving logic for out-of-memory errors.
Thanks @kleinsound
2016-01-16 16:37:35 -05:00
Brandon Amos 9d1bd99579 Update copyright notices.
sed -i "/[Cc]opyright.*2015.*Carnegie Mellon University/s/2015/2015-2016/;"  $(git grep '[Cc]opyright.*Carnegie Mellon University'  | sed 's/:.*//' | uniq)
2016-01-13 14:52:52 -05:00
Brandon Amos 8bbfd4da05 Training: Skip battch if nTripsFound == 0. 2016-01-13 14:32:43 -05:00
Brandon Amos dd5d0d11bd Training: Remove unused 'fbnn' import. 2016-01-12 17:30:06 -05:00
Brandon Amos c5b0accd4d Training: Add conditionals for cuda portions. 2016-01-11 16:17:36 -05:00
Brandon Amos 15cc5fef00 Training: Add CPU execution for tests. 2016-01-11 16:17:36 -05:00
Brandon Amos 8155840989 Training: Use e-lab's torch-toolbox sanitize. 2016-01-09 17:17:30 -05:00
Brandon Amos 6246fada83 Update code to nn4.v2 for #55. 2016-01-07 18:28:11 -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 df89d749d1 Clarify training procedure.
Summary of content from @Strateus and @melgor at
https://groups.google.com/forum/#!topic/cmu-openface/XgxfN8Xy9nA
2015-11-23 16:12:59 -05:00
Brandon Amos 2e8466afb1 Make suggestion 1 the default option. #48. 2015-11-09 15:58:05 -05:00
Brandon Amos aed184d860 Training: Suggestion 2 from @melgor in #48. 2015-11-08 17:08:06 -05:00
Brandon Amos ae348c22f9 Training: Fix error with tensors being unexpectedly doubles. 2015-11-06 17:09:47 -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 948bc2d3e9 Training: Update for_each call to apply. 2015-10-26 17:41:46 -04:00
Brandon Amos 69586648e6 Refactor: facenet->openface. 2015-10-10 21:28:41 -04:00
Brandon Amos d8790408d1 Initial commit of training code from development repo. 2015-09-23 19:49:45 -04:00