bambezius commited on
Commit
492a776
·
verified ·
1 Parent(s): 8f5f675

Upload folder using huggingface_hub

Browse files
Files changed (4) hide show
  1. config.json +13 -0
  2. metrics.json +11 -0
  3. model_info.json +6 -0
  4. model_params.joblib +3 -0
config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "prev_hidden_dims": 16,
3
+ "curr_hidden_dims": 64,
4
+ "time_embedding_dims": 8,
5
+ "lstm_embedding_dims": 128,
6
+ "dropout": 0.2,
7
+ "num_layers": 2,
8
+ "num_heads": 8,
9
+ "batch_size": 8,
10
+ "learning_rate": "1e-4",
11
+ "weight_decay": "1e-4",
12
+ "grad_accum_steps": 16
13
+ }
metrics.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "misc/epoch": 365,
3
+ "misc/patience": 8,
4
+ "train/loss": 0.8899281024932861,
5
+ "val/loss": 0.815583348274231,
6
+ "val/top1_accuracy": 0.7163023352622986,
7
+ "val/top2_accuracy": 0.880464494228363,
8
+ "val/top3_accuracy": 0.9355646371841431,
9
+ "val/mrr": 0.8296687006950378,
10
+ "misc/grad": 66.58882141113281
11
+ }
model_info.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "model_class": "TgnSymDynamic",
3
+ "framework": "jax",
4
+ "timestamp": "20260222_030505",
5
+ "model_name": "tgn-sym-dynamic-rsp"
6
+ }
model_params.joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8272302f931716dbf1286dff20c86607acde355e39efad2d4f50daa0b24eb135
3
+ size 1016760