Reduce time to reduce boundary errors

jloveric · Feb 16, 2023 · 84b6a25 · 84b6a25
1 parent eec6f4f
commit 84b6a25
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/config/euler.yaml b/config/euler.yaml
@@ -9,7 +9,7 @@ mlp:
   n_hidden: null
   resnet: False
   # We don't need this if we use max_abs normalization (it seems)
-  periodicity: 2.0
+  periodicity: null #2.0
 
   rescale_output: False
   scale: 2.0
@@ -45,7 +45,7 @@ time_decay: 0.0
 # and space derivatives. X is [-1, 1] and t is [0, 1] so t has
 # a scale factor of 0.5
 scale_x: 1
-scale_t: 0.5
+scale_t: 0.25
 
 # pde weighting
 loss_weight:

diff --git a/examples/high_order_euler.py b/examples/high_order_euler.py
@@ -55,7 +55,9 @@ def run(cfg: DictConfig):
             ):
 
                 trainer = Trainer(
-                    max_epochs=cfg.refinement.epochs,
+                    max_epochs=cfg.refinement.epochs
+                    if order < cfg.refinement.target_n
+                    else cfg.max_epochs,
                     gpus=cfg.gpus,
                     callbacks=[checkpoint_callback, ImageSampler(cfg=cfg), lr_monitor],
                 )