From a1e48c3460c6ca36034b7a17482b01b432bab1ac Mon Sep 17 00:00:00 2001
From: Matteo Bunino <matteo.bunino@gmail.com>
Date: Tue, 7 Nov 2023 13:23:44 +0100
Subject: [PATCH] UPDATE: reorg

---
 {src/itwinai => use-cases/mnist}/torch/README.md | 0
 use-cases/mnist/torch/pipeline.yaml              | 2 +-
 2 files changed, 1 insertion(+), 1 deletion(-)
 rename {src/itwinai => use-cases/mnist}/torch/README.md (100%)

diff --git a/src/itwinai/torch/README.md b/use-cases/mnist/torch/README.md
similarity index 100%
rename from src/itwinai/torch/README.md
rename to use-cases/mnist/torch/README.md
diff --git a/use-cases/mnist/torch/pipeline.yaml b/use-cases/mnist/torch/pipeline.yaml
index df29c3fe..9bb7fb98 100644
--- a/use-cases/mnist/torch/pipeline.yaml
+++ b/use-cases/mnist/torch/pipeline.yaml
@@ -25,7 +25,7 @@ executor:
             batch_size: 32
             pin_memory: True
             shuffle: False
-          epochs: 2
+          epochs: 30
           train_metrics:
             accuracy:
               class_path: torchmetrics.classification.MulticlassAccuracy