Cuong2004 commited on
Commit
51fdeea
·
verified ·
1 Parent(s): 1cc8d36

Add provenance sidecar for BiLSTM_T1.11_rel_2d

Browse files
checkpoints/best_BiLSTM_T1.11_rel_2d.provenance.json ADDED
@@ -0,0 +1,102 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "BiLSTM_T1.11_rel_2d",
3
+ "model_class": "BiLSTMModel",
4
+ "feature_method": "rel_2d",
5
+ "augment_method": "none",
6
+ "seed": 42,
7
+ "git_sha": "unknown",
8
+ "git_dirty": true,
9
+ "saved_timestamp": "2026-10-04 12:53:23 UTC",
10
+ "epoch": 31,
11
+ "total_epochs": 100,
12
+ "early_stopping_metric": "val_macro_f1",
13
+ "best_metric_value": 0.7461796074028558,
14
+ "val_loss": 1.7549551206037222,
15
+ "val_acc": 0.7595180722891566,
16
+ "val_macro_f1": 0.7461796074028558,
17
+ "train_loss": 0.0775666027426284,
18
+ "train_acc": 0.9737362971985384,
19
+ "hyperparameters": {
20
+ "lr": 0.001,
21
+ "weight_decay": 0.0001,
22
+ "optimizer": "AdamW",
23
+ "scheduler": "LambdaLR",
24
+ "loss_type": "ce",
25
+ "focal_gamma": 2.0,
26
+ "label_smoothing": 0.0,
27
+ "patience": 10,
28
+ "use_amp": true,
29
+ "amp_dtype": "torch.bfloat16"
30
+ },
31
+ "normalization_stats": {
32
+ "mean_shape": [
33
+ 1,
34
+ 26
35
+ ],
36
+ "mean_norm": 0.7991983294487,
37
+ "std_norm": 0.8995318412780762,
38
+ "is_normalized": true,
39
+ "seed": 42
40
+ },
41
+ "cmd_args": {
42
+ "command": "train",
43
+ "model": "BiLSTM",
44
+ "feature": "rel_2d",
45
+ "augment": "none",
46
+ "zero_frame": "interpolate",
47
+ "loss": "ce",
48
+ "focal_gamma": 2.0,
49
+ "label_smoothing": 0.0,
50
+ "exp_id": "T1.11",
51
+ "smoke_test": false,
52
+ "smoke_class": "barbell biceps curl",
53
+ "epochs": 100,
54
+ "batch_size": 16,
55
+ "lr": 0.001,
56
+ "weight_decay": 0.0001,
57
+ "patience": 10,
58
+ "seq_len": 32,
59
+ "train_stride": 16,
60
+ "val_test_stride": 32,
61
+ "use_class_weights": false,
62
+ "device": "cuda",
63
+ "metadata": "Final_dataset_metadata.csv",
64
+ "landmark_dir": "data/landmarks",
65
+ "output_dir": "outputs",
66
+ "checkpoint_dir": "/marimo/Gym_Classification/checkpoints",
67
+ "hidden_dim": null,
68
+ "num_layers": null,
69
+ "nhead": 8,
70
+ "dropout": null,
71
+ "seed": 42,
72
+ "num_workers": 0,
73
+ "early_stopping_metric": "val_macro_f1",
74
+ "optimizer": "adamw",
75
+ "scheduler": "cosine_warmup",
76
+ "warmup_epochs": 5,
77
+ "use_amp": true,
78
+ "amp_dtype": "bfloat16",
79
+ "in_memory": true,
80
+ "push_to_hf": true,
81
+ "hf_repo": "Cuong2004/gym-exercise-classification",
82
+ "report_file": "outputs/EXPERIMENT_RESULTS.md",
83
+ "video_level": true,
84
+ "checkpoint_name": null,
85
+ "force_retrain": true,
86
+ "no_test_eval": false,
87
+ "transformer_variant": "standard"
88
+ },
89
+ "dataset_info": {
90
+ "metadata_path": "Final_dataset_metadata.csv",
91
+ "landmark_dir": "data/landmarks",
92
+ "train_windows": 13136,
93
+ "val_windows": 2075,
94
+ "test_windows": 2743,
95
+ "seq_len": 32,
96
+ "train_stride": 16,
97
+ "val_test_stride": 32,
98
+ "zero_frame_strategy": "interpolate"
99
+ },
100
+ "checkpoint_file": "best_BiLSTM_T1.11_rel_2d.pt",
101
+ "checkpoint_sha256": "75bd6b5b236dc9744f64bf45d603ede537ed03f945e8dcb74b58fdb87f655cad"
102
+ }