limit action values in sampling stage

This commit is contained in:
NM512
2024-01-05 11:42:45 +09:00
parent a9e85e8b7c
commit a27711ab96
3 changed files with 33 additions and 15 deletions

View File

@@ -239,9 +239,10 @@ class ImagBehavior(nn.Module):
"learned",
config.actor["min_std"],
config.actor["max_std"],
config.actor["temp"],
absmax=1.0,
temp=config.actor["temp"],
unimix_ratio=config.actor["unimix_ratio"],
outscale=1.0,
outscale=config.actor["outscale"],
name="Actor",
)
self.value = networks.MLP(