Config

SciML · May 17, 2022 · 57a7abb · 57a7abb
1 parent ff9462c
commit 57a7abb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/examples/cifar10/options.jl b/examples/cifar10/options.jl
@@ -43,7 +43,7 @@ function parse_commandline_arguments()
         "--start-epoch"
             help = "manual epoch number (useful on restarts)"
             arg_type = Int
-            default = 0
+            default = 1
         "--print-freq"
             help = "print frequency"
             arg_type = Int

diff --git a/examples/src/config.jl b/examples/src/config.jl
@@ -42,7 +42,7 @@ function get_default_experiment_configuration(::Val{:CIFAR10}, ::Val{:TINY})
         abstol=5.0f-2,
         reltol=5.0f-2,
         ode_solver=VCABM3(),
-        pretrain_epochs=8,
+        pretrain_epochs=3,
         lr_scheduler=:COSINE,
         optimiser=:ADAM,
         eta=0.001f0 * scaling_factor(),
@@ -80,7 +80,7 @@ function get_default_experiment_configuration(::Val{:CIFAR10}, ::Val{:LARGE})
         abstol=5.0f-2,
         reltol=5.0f-2,
         ode_solver=VCABM3(),
-        pretrain_epochs=13,
+        pretrain_epochs=3,
         lr_scheduler=:COSINE,
         optimiser=:ADAM,
         eta=0.001f0 * scaling_factor(),