improved eval plotting functions

DAI-Lab · Oct 15, 2024 · 22b91d8 · 22b91d8
1 parent 3dd592a
commit 22b91d8
Show file tree

Hide file tree

Showing 13 changed files with 116 additions and 305 deletions.
diff --git a/config/model_config.yaml b/config/model_config.yaml
@@ -1,10 +1,10 @@
-device: 1 # 0, cpu
+device: 0 # 0, cpu
 seq_len: 96 # should not be changed for the current datasets
 input_dim: 2 # or 1 depending on user, but is dynamically set
 noise_dim: 256
 cond_emb_dim: 64
 shuffle: True
-sparse_conditioning_loss_weight: 0.5 # sparse conditioning training sample weight for loss computation [0, 1]
+sparse_conditioning_loss_weight: 0.8 # sparse conditioning training sample weight for loss computation [0, 1]
 freeze_cond_after_warmup: False # specify whether to freeze conditioning module parameters after warmup epochs
 save_cycle: 200 # specify number of epochs to save model after
 

diff --git a/docs/Makefile b/docs/Makefile
diff --git a/docs/authors.rst b/docs/authors.rst
diff --git a/docs/conf.py b/docs/conf.py
diff --git a/docs/contributing.rst b/docs/contributing.rst
diff --git a/docs/history.rst b/docs/history.rst
diff --git a/docs/images/dai-logo-white-200.png b/docs/images/dai-logo-white-200.png
diff --git a/docs/images/dai-logo-white.ico b/docs/images/dai-logo-white.ico
diff --git a/docs/index.rst b/docs/index.rst
diff --git a/docs/make.bat b/docs/make.bat
diff --git a/docs/readme.rst b/docs/readme.rst