Commit Graph

416 Commits

Author SHA1 Message Date
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
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
2caa6bc8f9 Added empty python packages
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-04 16:52:42 +01:00
227a1c9d8b Added src directory
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-04 16:09:25 +01:00