match target entropy hil serl

Co-authored-by: s1lent4gnt <kmeftah.khalil@gmail.com>
This commit is contained in:
AdilZouitine
2025-04-15 08:00:38 +00:00
committed by Michel Aractingi
parent 5c352ae558
commit 8122721f6d
3 changed files with 11 additions and 7 deletions

View File

@@ -231,6 +231,7 @@ def act_with_policy(
cfg=cfg.policy,
env_cfg=cfg.env,
)
policy = policy.eval()
assert isinstance(policy, nn.Module)
obs, info = online_env.reset()