UPDATE: reorg

interTwin-eu · Nov 7, 2023 · a1e48c3 · a1e48c3
1 parent 8dbcfcc
commit a1e48c3
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/src/itwinai/torch/README.md → use-cases/mnist/torch/README.md b/src/itwinai/torch/README.md → use-cases/mnist/torch/README.md
diff --git 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