bug fix for onehot distribution

2024-01-14 21:55:34 +09:00
parent 5abaf056ad
commit a4fdfad938
2 changed files with 6 additions and 6 deletions
--- a/models.py
+++ b/models.py
@@ -228,7 +228,7 @@ class ImagBehavior(nn.Module):
            config.act,
            config.norm,
            config.actor["dist"],
-            "learned",
+            config.actor["std"],
            config.actor["min_std"],
            config.actor["max_std"],
            absmax=1.0,