(self, obs: torch.Tensor, proprio_hist: torch.Tensor)
| 137 | self.rms_hist = rms_hist |
| 138 | |
| 139 | def forward(self, obs: torch.Tensor, proprio_hist: torch.Tensor) -> torch.Tensor: |
| 140 | obs_n = self.rms_obs(obs) |
| 141 | hist_n = self.rms_hist(proprio_hist) |
| 142 | mu = self.model.act_inference({"obs": obs_n, "proprio_hist": hist_n}) |
| 143 | return torch.clamp(mu, -1.0, 1.0) |
| 144 | |
| 145 | |
| 146 | class Stage1ExportWrapper(torch.nn.Module): |
nothing calls this directly
no test coverage detected