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