Honglong Tian

[WIP] Added pred-continue and collector-mask for dreamer

1

## Description Added optional `pred-continue` function like https://github.com/danijar/dreamer/blob/master/dreamer.py#L187C32-L187C32, and using `collector-mask` to remove empty data. ## Motivation and Context To make sure dreamer can work on early-stop env. - [...

## Describe the bug When using `autocast`` in dreamer example, there is a RuntimeError: `` RuntimeError: masked_scatter_: expected self and source to have same dtypes but gotHalf and Float ``...

[BUG] dreamer example broken caused by ``ObservationNorm``

2

## Describe the bug ``ObservationNorm`` with fixed ``obs_norm_state_dict`` is used in dreamer example, and this ``obs_norm_state_dict`` is estimated by only `cfg.init_env_steps` steps rollout. When the observation is ``pixels``, some pixels...

[BUG] Planning (`MPCPlannerBase`) should consider `done`

9

## Describe the bug In the current implementation, all subclasses of `MPCPlannerBase` do not consider `done` thrown by env during the planning process, which means that MPC is invalid in...