| { | |
| "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 | |
| } | |
| } | |
| } |