Hugging Face's logo

cesarali
/
StudyNodePK_cluster

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

best val_rmse 0.0141

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.008973164483904839,
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": 25,
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": 0.0001,
194
  "log_image_every_epoch": 5,
195
  "log_interval": 1,
196
  "log_vcp": true,
 
1
  {
2
+ "best_val_loss": 0.014130729250609875,
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": 27,
165
  "tags": [
166
  "FAttention",
167
  "Long",
 
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,
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0506b265a430ee9af49062bd616477c1aa2ee43645833fb24f80072a96830f90
3
  size 10189543
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf65fbdf180467fcf4c0f5e938e68cb0d765ab795e3b4f893b80918e6eb756e5
3
  size 10189543