diff --git a/modules/deeplearning/cloudheight.py b/modules/deeplearning/cloudheight.py index 1728c0762f7f9e887ad779d1973f19b4af514f37..fc0c3f112673f89f89b20d4991efc0a2ca9db6dc 100644 --- a/modules/deeplearning/cloudheight.py +++ b/modules/deeplearning/cloudheight.py @@ -16,9 +16,9 @@ CACHE_DATA_IN_MEM = True CACHE_GFS = True DISK_CACHE = True -PROC_BATCH_SIZE = 40 +PROC_BATCH_SIZE = 60 NumLabels = 1 -BATCH_SIZE = 512 +BATCH_SIZE = 1024 NUM_EPOCHS = 200 GLOBAL_BATCH_SIZE = BATCH_SIZE * 3 @@ -687,10 +687,10 @@ class CloudHeightNN: labels = mini_batch[2] pred = self.model(inputs, training=False) t_loss = self.loss(labels, pred) - t_loss = tf.nn.compute_average_loss(t_loss, global_batch_size=GLOBAL_BATCH_SIZE) self.test_loss(t_loss) self.test_accuracy(labels, pred) + return t_loss @tf.function