[Port HIL_SERL] Final fixes for the Reward Classifier (#598)

This commit is contained in:
Eugene Mironov
2025-01-06 17:34:00 +07:00
committed by Michel Aractingi
parent e5801f467f
commit d1d6ffd23c
10 changed files with 7780 additions and 15 deletions
@@ -333,7 +333,6 @@ class Critic(nn.Module):
value = self.output_layer(x)
return value.squeeze(-1)
class Policy(nn.Module):
def __init__(
self,