From ea1b4943cfec5c99faf084b786fb1dd62744a8ea Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Fri, 8 Feb 2019 18:12:33 +0100 Subject: [PATCH] Fixed function call Signed-off-by: Jim Martens --- src/twomartens/masterthesis/aae/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/aae/train.py b/src/twomartens/masterthesis/aae/train.py index 70a4154..a5743cc 100644 --- a/src/twomartens/masterthesis/aae/train.py +++ b/src/twomartens/masterthesis/aae/train.py @@ -564,7 +564,7 @@ if __name__ == "__main__": tf.enable_eager_execution() inlier_classes = [0] iteration = 1 - train_dataset, _ = prepare_training_data(test_fold_id=0, inlier_classes=[0], + train_dataset, _ = prepare_training_data(test_fold_id=0, inlier_classes=inlier_classes, total_classes=10) train_summary_writer = summary_ops_v2.create_file_writer( './summaries/train/number-' + str(inlier_classes[0]) + '/' + str(iteration))