Hugging Face's logo

cesarali
/
StudyNodePK_cluster

PyTorch
English
generative-pk
node_pk
predictive
Model card Files
xet
Community
cesarali commited on
Commit
eb50fa3
·
verified ·
1 Parent(s): 76d1bc8

best val_rmse 0.0075

Browse files
Files changed (2) hide show
  1. config.json +3 -3
  2. pytorch_model.bin +1 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "best_val_loss": 0.009428302757441998,
3
  "comet_ai_key": null,
4
  "context_observations": {
5
  "add_rem": true,
@@ -161,7 +161,7 @@
161
  "use_covariance": false,
162
  "zi_latent_dim": 128
163
  },
164
- "run_index": 43,
165
  "tags": [
166
  "FAttention",
167
  "Long",
@@ -190,7 +190,7 @@
190
  "epochs": 950,
191
  "eps": 1e-08,
192
  "gradient_clip_val": 1.0,
193
- "learning_rate": 1e-05,
194
  "log_image_every_epoch": 5,
195
  "log_interval": 1,
196
  "log_vcp": true,
 
1
  {
2
+ "best_val_loss": 0.007497346960008144,
3
  "comet_ai_key": null,
4
  "context_observations": {
5
  "add_rem": true,
 
161
  "use_covariance": false,
162
  "zi_latent_dim": 128
163
  },
164
+ "run_index": 41,
165
  "tags": [
166
  "FAttention",
167
  "Long",
 
190
  "epochs": 950,
191
  "eps": 1e-08,
192
  "gradient_clip_val": 1.0,
193
+ "learning_rate": 0.0001,
194
  "log_image_every_epoch": 5,
195
  "log_interval": 1,
196
  "log_vcp": true,
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:adfb906f44d61acaa3ca6c522a5629e5b5ed97e9357c192d63f56e8a8aa62e9b
3
  size 23990293
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:339b1d7fe55b370a631f42d85aa31edbf56f7f14009b31495e2695209578742c
3
  size 23990293