backup wip

This commit is contained in:
Alexander Soare
2024-04-15 19:06:44 +01:00
parent 14f3ffb412
commit 5608e659e6
14 changed files with 1059 additions and 977 deletions

View File

@@ -163,7 +163,8 @@ class ActionChunkingTransformerPolicy(nn.Module):
@torch.no_grad
def select_action(self, batch: dict[str, Tensor], **_) -> Tensor:
"""
"""Select a single action given environment observations.
This method wraps `select_actions` in order to return one action at a time for execution in the
environment. It works by managing the actions in a queue and only calling `select_actions` when the
queue is empty.