_target_: torch.nn.Identity
output_dim: ${env.obs_dim}