Upload model_config.json with huggingface_hub
Browse files- model_config.json +7 -7
model_config.json
CHANGED
|
@@ -1,11 +1,11 @@
|
|
| 1 |
{
|
| 2 |
"model_info": {
|
| 3 |
"latent_dims": 128,
|
| 4 |
-
"compression_mode": "
|
| 5 |
"model_type": "SimpleAE",
|
| 6 |
-
"input_dim":
|
| 7 |
"layer_sizes": [
|
| 8 |
-
|
| 9 |
8192,
|
| 10 |
4096,
|
| 11 |
2048,
|
|
@@ -45,14 +45,14 @@
|
|
| 45 |
"tracerx_priority_weight": 2.0
|
| 46 |
},
|
| 47 |
"training_results": {
|
| 48 |
-
"best_epoch":
|
| 49 |
-
"best_validation_loss": 0.
|
| 50 |
-
"final_epoch":
|
| 51 |
"training_completed": true
|
| 52 |
},
|
| 53 |
"system_info": {
|
| 54 |
"device": "cuda:1",
|
| 55 |
-
"timestamp": "2025-08-
|
| 56 |
"python_version": "3.10.12",
|
| 57 |
"pytorch_version": "2.6.0+cu124"
|
| 58 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"model_info": {
|
| 3 |
"latent_dims": 128,
|
| 4 |
+
"compression_mode": "samples",
|
| 5 |
"model_type": "SimpleAE",
|
| 6 |
+
"input_dim": 954,
|
| 7 |
"layer_sizes": [
|
| 8 |
+
954,
|
| 9 |
8192,
|
| 10 |
4096,
|
| 11 |
2048,
|
|
|
|
| 45 |
"tracerx_priority_weight": 2.0
|
| 46 |
},
|
| 47 |
"training_results": {
|
| 48 |
+
"best_epoch": 0,
|
| 49 |
+
"best_validation_loss": 0.00013786639667526932,
|
| 50 |
+
"final_epoch": 0,
|
| 51 |
"training_completed": true
|
| 52 |
},
|
| 53 |
"system_info": {
|
| 54 |
"device": "cuda:1",
|
| 55 |
+
"timestamp": "2025-08-26T21:20:03.230274",
|
| 56 |
"python_version": "3.10.12",
|
| 57 |
"pytorch_version": "2.6.0+cu124"
|
| 58 |
}
|