From 282c1f86cc3c2a9f609b48127c4544d3743e02b4 Mon Sep 17 00:00:00 2001
From: Daniel Franco <danialtza@gmail.com>
Date: Mon, 14 Aug 2023 14:14:11 +0200
Subject: [PATCH] Update 3D SR template

---
 .../super-resolution/3d_super-resolution.yaml    | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/templates/super-resolution/3d_super-resolution.yaml b/templates/super-resolution/3d_super-resolution.yaml
index c157ed87..230e8cdb 100644
--- a/templates/super-resolution/3d_super-resolution.yaml
+++ b/templates/super-resolution/3d_super-resolution.yaml
@@ -3,25 +3,24 @@ SYSTEM:
 
 PROBLEM:
     TYPE: SUPER_RESOLUTION
-    NDIM: 2D
+    NDIM: 3D
     SUPER_RESOLUTION:
         UPSCALING: 2
   
 DATA: 
-    PATCH_SIZE: (48, 48, 1)
+    PATCH_SIZE: (10, 48, 48, 1)
     TRAIN:                                                                                                              
         PATH: /path/to/data
         GT_PATH: /path/to/data
         IN_MEMORY: True
     VAL:
-        FROM_TRAIN: True
         SPLIT_TRAIN: 0.1
     TEST:                                                                                                               
         IN_MEMORY: True
         LOAD_GT: True
         PATH: /path/to/data
         GT_PATH: /path/to/data
-        PADDING: (8,8)
+        PADDING: (2,8,8)
 
 AUGMENTOR:
     ENABLE: True
@@ -31,17 +30,20 @@ AUGMENTOR:
     ZFLIP: True
 
 MODEL:
-    ARCHITECTURE: edsr
+    ARCHITECTURE: unet
     LOAD_CHECKPOINT: False
 
 TRAIN:
     ENABLE: True
     OPTIMIZER: ADAMW
     LR: 1.E-4
-    BATCH_SIZE: 24
-    EPOCHS: 360
+    BATCH_SIZE: 12
+    EPOCHS: 400
     PATIENCE: 50
   
 TEST:
     ENABLE: True
     AUGMENTATION: False
+    STATS:
+        PER_PATCH: True
+        MERGE_PATCHES: True