File size: 971 Bytes
185fee0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 | {
"model_class": "ActionAutoencoder",
"total_params": 11790537,
"trainable_params": 9306553,
"total_params_M": 11.7905,
"trainable_params_M": 9.3066,
"modules": {
"encoder": {
"total_params": 2483984,
"trainable_params": 2483984,
"total_params_M": 2.484,
"trainable_params_M": 2.484
},
"decoder": {
"total_params": 2219784,
"trainable_params": 2219784,
"total_params_M": 2.2198,
"trainable_params_M": 2.2198
},
"dynamics_model": {
"total_params": 3496976,
"trainable_params": 3496976,
"total_params_M": 3.497,
"trainable_params_M": 3.497
},
"progress_model": {
"total_params": 1105425,
"trainable_params": 1105425,
"total_params_M": 1.1054,
"trainable_params_M": 1.1054
},
"target_encoder": {
"total_params": 2483984,
"trainable_params": 0,
"total_params_M": 2.484,
"trainable_params_M": 0.0
}
}
} |