network: GMMHead

network_kwargs:
    hidden_size: 1024
    num_layers: 2
    min_std: 0.0001
    num_modes: 5
    low_eval_noise: false
    activation: "softplus"

loss_kwargs:
    loss_coef: 1.0
