File size: 417 Bytes
dad54c9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
dynamics_model:
_target_: mbrl.models.GaussianMLP
num_layers: 3
ensemble_size: 5
device: cpu
hid_size: 200
in_size: 5
out_size: 4
deterministic: false
propagation_method: fixed_model
activation_fn_cfg:
_target_: torch.nn.LeakyReLU
negative_slope: 0.01
algorithm:
learned_rewards: false
target_is_delta: true
normalize: true
overrides:
model_batch_size: 32
validation_ratio: 0.05
|