Hugging Face's logo

cesarali
/
StudyNodePK_cluster

PyTorch
English
generative-pk
node_pk
predictive
Model card Files
xet
Community
cesarali commited on
Commit
655ed81
·
verified ·
1 Parent(s): 170a81a

best val_rmse 0.0081

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.008182710967957973,
3
  "comet_ai_key": null,
4
  "context_observations": {
5
  "add_rem": true,
@@ -161,7 +161,7 @@
161
  "use_covariance": false,
162
  "zi_latent_dim": 256
163
  },
164
- "run_index": 33,
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.008130740374326706,
3
  "comet_ai_key": null,
4
  "context_observations": {
5
  "add_rem": true,
 
161
  "use_covariance": false,
162
  "zi_latent_dim": 256
163
  },
164
+ "run_index": 35,
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:f35f949c3ecbe2fe329b98925cc35159889f1cd55debb3a847ed0e0e217b42af
3
  size 24252949
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:634292f2475f508aa810e3a39d7d5ff0f29691b5ea1fdffd86b4cdf482701b28
3
  size 24252949