EmbarkStudios / EmbarkStudios/emote

Add epsilon-size override for ONNX exporter

Open
#170 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
29
Forks
0
PR merge metrics
No merged PRs in 30d

Description

I'll make a note to augment the ONNX exporter so it can handle this; this feels like it could end up in other places too.

```py3
def forward(self, obs: Tensor, epsilon: Tensor = None):
# we need to discard the extra dimensions of epsilon.
# the input epsilon is given for the original action space
# however, the policy outputs latent actions.
if epsilon is not None:
epsilon = epsilon[:, : self.latent_size]
```
_Originally posted by @tgolsson in https://github.com/EmbarkStudios/emote/pull/159#discussion_r1325550640_

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.