shivakanthsujit's picture
Init Commit
987de76
raw
history blame
417 Bytes
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