Use original_action_dim instead of self.config.max_action_dim when processing losses to exclude padded regions. Including padding areas in loss computation would impair the model's action reconstruction capability. |
||
---|---|---|
.. | ||
datasets | ||
envs | ||
optim | ||
policies | ||
robot_devices | ||
utils | ||
constants.py |