Hugging Face's logo

cesarali
/
ContextVAENodePK_cluster

PyTorch
English
generative-pk
node_pk
generative
Model card Files
xet
Community
cesarali commited on
Commit
4365ff6
·
verified ·
1 Parent(s): 0a570e1

best val_rmse 0.2250

Browse files
Files changed (2) hide show
  1. config.json +5 -2
  2. pytorch_model.bin +1 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "best_val_loss": 0.20946753025054932,
3
  "comet_ai_key": null,
4
  "context_observations": {
5
  "add_rem": true,
@@ -160,6 +160,9 @@
160
  "time_obs_encoder_hidden_dim": 128,
161
  "time_obs_encoder_output_dim": 128,
162
  "use_attention": true,
 
 
 
163
  "zi_latent_dim": 200
164
  },
165
  "run_index": 0,
@@ -185,7 +188,7 @@
185
  0.9,
186
  0.999
187
  ],
188
- "epochs": 6,
189
  "eps": 1e-08,
190
  "gradient_clip_val": 1.0,
191
  "learning_rate": 0.0001,
 
1
  {
2
+ "best_val_loss": 0.22500772774219513,
3
  "comet_ai_key": null,
4
  "context_observations": {
5
  "add_rem": true,
 
160
  "time_obs_encoder_hidden_dim": 128,
161
  "time_obs_encoder_output_dim": 128,
162
  "use_attention": true,
163
+ "use_kl_i": true,
164
+ "use_kl_init": true,
165
+ "use_kl_s": true,
166
  "zi_latent_dim": 200
167
  },
168
  "run_index": 0,
 
188
  0.9,
189
  0.999
190
  ],
191
+ "epochs": 3,
192
  "eps": 1e-08,
193
  "gradient_clip_val": 1.0,
194
  "learning_rate": 0.0001,
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:033c6653c4ec0093c13b1427475a2d278a99cf00cb3271026065ac8fd0cfde7b
3
  size 12058855
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b33d17e7726075c59f33eee2b502fb00fedb747db6db5f028c8fdc77b24861ee
3
  size 12058855