From be834ce3a0c9bd4bc834d46dc15daaff45eeecf3 Mon Sep 17 00:00:00 2001
From: Luc Giffon <luc.giffon@lis-lab.fr>
Date: Mon, 12 Nov 2018 13:36:14 +0100
Subject: [PATCH] solve issue count batch
---
.../deepstrom_classif_end_to_end.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/main/experiments/scripts/november_2018/end_to_end_with_2_layers_only_dense_with_augment/deepstrom_classif_end_to_end.py b/main/experiments/scripts/november_2018/end_to_end_with_2_layers_only_dense_with_augment/deepstrom_classif_end_to_end.py
index 9fe8402..9af6985 100644
--- a/main/experiments/scripts/november_2018/end_to_end_with_2_layers_only_dense_with_augment/deepstrom_classif_end_to_end.py
+++ b/main/experiments/scripts/november_2018/end_to_end_with_2_layers_only_dense_with_augment/deepstrom_classif_end_to_end.py
@@ -272,11 +272,12 @@ def main(paraman, resman, printman):
if j % 100 == 0:
logger.info(
"epoch: {}/{}; batch: {}/{}; batch_shape: {}; loss: {}; acc: {}".format(i, paraman["--num-epoch"],
- j + 1, int(data.train[0].shape[0] / paraman["--batch-size"]) + 1,
+ k, int(data.train[0].shape[0] / paraman["--batch-size"]) + 1,
X_batch.shape, loss,
acc))
if paraman["--tensorboard"]:
summary_writer.add_summary(summary_str, j)
+ k += 1
j += 1
logger.info("Evaluation on validation data")
--
GitLab