Commit Graph

522 Commits

Author SHA1 Message Date
422e01a917 Improved docstring in aae package
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 22:23:24 +01:00
85f3100b5d Updated docstring for model module
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 21:57:57 +01:00
81f245be99 Updated docstring for util module
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 21:22:00 +01:00
9bf7229b1e Added TODOs to train module
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 21:14:45 +01:00
90fa27d9ef Replaced all binary cross entropy losses with log_loss
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 21:01:38 +01:00
a51b535728 Improved naming for histogram summaries and added variable summaries
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 20:47:34 +01:00
0cbab44f01 Fixed summaries for histograms
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 20:25:56 +01:00
a319cf988e Added summaries for grads and learning rate
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 20:13:21 +01:00
e85b850c63 Fixed frequency of summary savings
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 19:41:08 +01:00
a8f8015ad8 Improved summary logs
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 19:27:33 +01:00
a2aa794d47 Fixed sample image summary
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 18:58:19 +01:00
3b9742a1b4 Improved summary logging
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 18:40:41 +01:00
75d7c769c7 Made early stopping conditional and turned it off
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 18:22:25 +01:00
ea1b4943cf Fixed function call
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 18:12:33 +01:00
d9a35893af Added epoch counter to restored variables
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 18:12:02 +01:00
1d4f56e40d Improved docstrings
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 17:12:41 +01:00
028513b404 Renamed internal functions to make them protected
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 16:38:58 +01:00
483cb4eb3e Removed obsolete function
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 16:35:00 +01:00
2d1fee8048 Moved training of epoch into separate function
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 16:34:23 +01:00
6f36aa7faf Added more parameters to checkpoint
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 15:45:27 +01:00
658973d90d Added weights prefix to parameters
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 15:34:49 +01:00
88ffd4f879 Moved dataset preparation into own function
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 15:32:01 +01:00
8e0a063759 Removed abspath usage in specifying checkpoint dir
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 15:03:47 +01:00
279e848434 Fixed specification of checkpoint dir
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 14:59:50 +01:00
720ff63c7d Updated saving of checkpoints
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 14:53:45 +01:00
3c5f332880 Added saving of weights after each epoch
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 14:22:27 +01:00
ec9b64a757 Prevented unnecessary early stops
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 13:51:41 +01:00
948ef64423 Added formatting of decoder loss
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 13:41:09 +01:00
a0094266b3 Added training iteration variable to train function
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 13:36:09 +01:00
3bde913323 Extended early stopping with total loss
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 13:27:19 +01:00
c19ff19820 Implemented early stopping
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 13:20:17 +01:00
682e11b435 Added step values to image summaries
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 12:41:10 +01:00
d84b062bf9 Fixed image summaries
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 12:36:36 +01:00
01db450032 Added image summaries
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 12:24:07 +01:00
7adb703292 Removed the actual image saving and improved docstrings
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 12:23:39 +01:00
fc1aa20396 Improved saving location of summaries
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 11:37:35 +01:00
5ba3bc552f Removed caching of dataset
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 11:37:05 +01:00
85d743c058 Added inlier class to saved weights
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 10:46:25 +01:00
ce6cda5351 Added comments to give context to the folds
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 10:46:00 +01:00
3f08c54679 Turned-off implicit summary saving at end of epoch
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 10:32:28 +01:00
9d0f93689e Improved paths for saved model weights
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 09:07:33 +01:00
cc5e7f89a2 Removed obsolete shuffling
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 08:51:43 +01:00
2112e1dd0c Added summary writer
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 08:39:59 +01:00
ab52f0feb0 Improved names of loss metrics
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 08:30:06 +01:00
16e4aaa348 Fixed loss logging
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 08:29:01 +01:00
758cdb5520 Moved to tf.data.Dataset API
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 08:28:41 +01:00
d9cd24f769 Added verbosity flag
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 07:06:16 +01:00
1041f24b99 Removed obsolete loss history code
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 07:01:38 +01:00
4dda663e3b Added resetting the states of the metrics
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 06:59:57 +01:00
a6e90e70b3 Added logging of loss values as summaries
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 06:51:40 +01:00