ready for review

This commit is contained in:
Alexander Soare
2024-04-12 11:36:52 +01:00
parent 5666ec3ec7
commit 6d0a45a97d
7 changed files with 11 additions and 42 deletions

View File

@@ -213,7 +213,7 @@ class ActionChunkingTransformerPolicy(nn.Module):
return action[: self.n_action_steps]
def __call__(self, *args, **kwargs) -> dict:
# TODO(now): Temporary bridge until we know what to do about the `update` method.
# TODO(alexander-soare): Temporary bridge until we know what to do about the `update` method.
return self.update(*args, **kwargs)
def _preprocess_batch(