|
|
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 |
|
|
|
65638974d3
|
Specified correct datatype for global step variable
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 06:48:13 +01:00 |
|
|
|
8aee9817f0
|
Removed obsolete code
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 06:38:56 +01:00 |
|
|
|
c82f5bf87c
|
Fixed docstrings
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 06:38:27 +01:00 |
|
|
|
14cb27afd7
|
Extracted training steps of models into separate functions
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 06:37:45 +01:00 |
|
|
|
887435ddd7
|
Introduced global steps for each optimizer
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 05:55:19 +01:00 |
|
|
|
bb76a51d76
|
Switched to keras binary_crossentropy loss
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 05:42:54 +01:00 |
|
|
|
c52a3cdfb9
|
Added names and types to loss metrics
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-07 18:06:33 +01:00 |
|
|
|
cce0975a13
|
Realized learning rate decay with variable
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-07 17:55:51 +01:00 |
|
|
|
ebfa67abb0
|
Fixed saving loss history
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-07 17:55:24 +01:00 |
|
|
|
abf52a88b0
|
Properly created global step
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-07 17:29:37 +01:00 |
|
|
|
661e99c9f5
|
Added required __init__ file
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-07 16:58:50 +01:00 |
|
|
|
bfc888389f
|
Added missing dependencies and switched to GPU dependency
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-07 16:58:32 +01:00 |
|
|
|
ecbfbf2a2a
|
Added mnist training function for eager mode
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-07 16:57:45 +01:00 |
|
|
|
d4dc705dfb
|
Added utility functions to save prediction results as images
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-07 16:21:07 +01:00 |
|
|
|
ca31ae2fbc
|
Added mnist compatible models for AAE
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-07 15:58:24 +01:00 |
|