Add more options to the global config (#289)

theory-in-progress · web-flow · commit e039b478c2a4 · 2023-08-19T15:15:43.000+02:00
* Add more option to the global config

* Fix Lint

* Modify config on deselection

* Update tests configs

* Remove unnecessary default dist

* Export default config, removed unnecessary defaults

* Remove unused imports
diff --git a/src/store.js b/src/store.js
@@ -34,18 +34,28 @@ export const msg = reactive({
   color: 'red'
 })
 
+// default configs for the templates
+export const default_config = {
+  template: '',
+  include_test: false,
+  output_dir: './logs',
+  log_every_iters: 10,
+  logger: 'tensorboard',
+  save_training: true,
+  save_evaluation: true,
+  patience: 3,
+  filename_prefix: 'training',
+  save_every_iters: 1000,
+  n_saved: 2
+}
+
 // main reactive object
 // store.code - the final rendered code to be included in archive
 // store.config - the internal config to track user input
 // codeUrl - the wget url for integration and local development
 export const store = reactive({
   code: {},
-  config: {
-    template: '',
-    include_test: false,
-    output_dir: './logs',
-    log_every_iters: 10
-  },
+  config: default_config,
   codeUrl: ''
 })
 
diff --git a/src/tests/ci-configs/text-classification-launch.yaml b/src/tests/ci-configs/text-classification-launch.yaml
@@ -20,3 +20,7 @@ max_length: 256
 lr: 0.00005
 output_dir: ./logs
 log_every_iters: 2
+filename_prefix: training
+n_saved: 2
+save_every_iters: 1000
+patience: 3
diff --git a/src/tests/ci-configs/text-classification-simple.yaml b/src/tests/ci-configs/text-classification-simple.yaml
@@ -20,3 +20,7 @@ max_length: 256
 lr: 0.00005
 output_dir: ./logs
 log_every_iters: 2
+filename_prefix: training
+n_saved: 2
+save_every_iters: 1000
+patience: 3
diff --git a/src/tests/ci-configs/text-classification-spawn.yaml b/src/tests/ci-configs/text-classification-spawn.yaml
@@ -24,3 +24,7 @@ nproc_per_node: 2
 nnodes: 1
 output_dir: ./logs
 log_every_iters: 2
+filename_prefix: training
+n_saved: 2
+save_every_iters: 1000
+patience: 3
diff --git a/src/tests/ci-configs/vision-classification-launch.yaml b/src/tests/ci-configs/vision-classification-launch.yaml
@@ -12,3 +12,7 @@ debug: false
 model: resnet18
 output_dir: ./logs
 log_every_iters: 2
+filename_prefix: training
+n_saved: 2
+save_every_iters: 1000
+patience: 3
diff --git a/src/tests/ci-configs/vision-classification-simple.yaml b/src/tests/ci-configs/vision-classification-simple.yaml
@@ -12,3 +12,7 @@ debug: false
 model: resnet18
 output_dir: ./logs
 log_every_iters: 2
+filename_prefix: training
+n_saved: 2
+save_every_iters: 1000
+patience: 3
diff --git a/src/tests/ci-configs/vision-classification-spawn.yaml b/src/tests/ci-configs/vision-classification-spawn.yaml
@@ -16,3 +16,7 @@ nproc_per_node: 2
 nnodes: 1
 output_dir: ./logs
 log_every_iters: 2
+filename_prefix: training
+n_saved: 2
+save_every_iters: 1000
+patience: 3
diff --git a/src/tests/ci-configs/vision-dcgan-launch.yaml b/src/tests/ci-configs/vision-dcgan-launch.yaml
@@ -14,3 +14,7 @@ d_filters: 64
 g_filters: 64
 output_dir: ./logs
 log_every_iters: 2
+filename_prefix: training
+n_saved: 2
+save_every_iters: 1000
+patience: 3
diff --git a/src/tests/ci-configs/vision-dcgan-simple.yaml b/src/tests/ci-configs/vision-dcgan-simple.yaml
@@ -14,3 +14,7 @@ d_filters: 64
 g_filters: 64
 output_dir: ./logs
 log_every_iters: 2
+filename_prefix: training
+n_saved: 2
+save_every_iters: 1000
+patience: 3
diff --git a/src/tests/ci-configs/vision-dcgan-spawn.yaml b/src/tests/ci-configs/vision-dcgan-spawn.yaml
@@ -18,3 +18,7 @@ nproc_per_node: 2
 nnodes: 1
 output_dir: ./logs
 log_every_iters: 2
+filename_prefix: training
+n_saved: 2
+save_every_iters: 1000
+patience: 3
diff --git a/src/tests/ci-configs/vision-segmentation-launch.yaml b/src/tests/ci-configs/vision-segmentation-launch.yaml
@@ -13,3 +13,7 @@ accumulation_steps: 4
 num_classes: 21
 output_dir: ./logs
 log_every_iters: 2
+filename_prefix: training
+n_saved: 2
+save_every_iters: 1000
+patience: 3
diff --git a/src/tests/ci-configs/vision-segmentation-simple.yaml b/src/tests/ci-configs/vision-segmentation-simple.yaml
@@ -13,3 +13,7 @@ accumulation_steps: 4
 num_classes: 21
 output_dir: ./logs
 log_every_iters: 2
+filename_prefix: training
+n_saved: 2
+save_every_iters: 1000
+patience: 3
diff --git a/src/tests/ci-configs/vision-segmentation-spawn.yaml b/src/tests/ci-configs/vision-segmentation-spawn.yaml
@@ -17,3 +17,7 @@ nproc_per_node: 2
 nnodes: 1
 output_dir: ./logs
 log_every_iters: 2
+filename_prefix: training
+n_saved: 2
+save_every_iters: 1000
+patience: 3
diff --git a/src/views/Create.vue b/src/views/Create.vue
@@ -20,7 +20,7 @@ import PaneRight from '../components/PaneRight.vue'
 import PaneLeft from '../components/PaneLeft.vue'
 import Footer from '../components/Footer.vue'
 import { onUnmounted } from 'vue'
-import { store } from '../store'
+import { default_config, store } from '../store'
 
 export default {
   components: {
@@ -42,10 +42,9 @@ export default {
       // since we delete each property,
       // it is better to reassign the initial values
       // which are defined in store.js
-      store.config.template = ''
-      store.config.include_test = false
-      store.config.output_dir = './logs'
-      store.config.log_every_iters = 2
+      for (const key in default_config) {
+        store.config[key] = default_config[key]
+      }
     })
   }
 }
diff --git a/src/views/PageNotFound.vue b/src/views/PageNotFound.vue
@@ -19,7 +19,7 @@
 import NavBar from '../components/NavBar.vue'
 import Footer from '../components/Footer.vue'
 import { onUnmounted } from 'vue'
-import { store } from '../store'
+import { default_config, store } from '../store'
 
 export default {
   components: {
@@ -38,10 +38,9 @@ export default {
       // since we delete each property,
       // it is better to reassign the initial values
       // which are defined in store.js
-      store.config.template = ''
-      store.config.include_test = false
-      store.config.output_dir = './logs'
-      store.config.log_every_iters = 2
+      for (const key in default_config) {
+        store.config[key] = default_config[key]
+      }
     })
   }
 }
