Fix dense depth loss and model, fix metric naming

2021-09-13 20:35:59 +09:30
parent 28b11aaa44
commit a053238310
4 changed files with 20 additions and 4 deletions
--- a/train.py
+++ b/train.py
@@ -19,7 +19,7 @@ def compile(model, optimiser=keras.optimizers.SGD(), loss=keras.losses.MeanSquar
                  loss=loss,
                  metrics=[keras.metrics.RootMeanSquaredError(),
                           keras.metrics.MeanSquaredError(),
-                           delta1_metric,
+                           delta1,
                           delta2,
                           delta3,
                           keras.metrics.MeanAbsolutePercentageError(),