add script for vae

thatgeeman · Nov 30, 2024 · fb40a7a · fb40a7a
1 parent 8759987
commit fb40a7a
Show file tree

Hide file tree

Showing 7 changed files with 614 additions and 1,232 deletions.
diff --git a/.gitignore b/.gitignore
@@ -3,6 +3,7 @@ index_files/
 _proc/
 data/
 */*.pth
+outputs/
 
 *.bak
 .gitattributes

diff --git a/config/config.yaml b/config/config.yaml
@@ -2,7 +2,7 @@ defaults:
   - _self_
   - dataset: nyc_taxi
 
-base_dir: ..
+base_dir: .
 model_dir: ${base_dir}/models
 num_workers: 4
 device: cuda
@@ -21,7 +21,7 @@ kernel_size: 5 # kernel size would be (kernel_size, n_signals)
 stride: 2 # stride would be (stride, n_signals)
 # training
 lr: 1e-3
-epochs: 50
+epochs: 100
 wd: 1e-6
 # scheduler
 patience: 10
-Original file line number
+Diff line change
@@ Expand Up / @@ -3,6 +3,7 @@ index_files/ @@
     _proc/
     data/
     */*.pth
+    outputs/
     *.bak
     .gitattributes
@@ Expand Down @@