updated result, requirements and torch version

2023-03-24 07:51:57 +09:00
parent 2504426164
commit 942eae10a9
6 changed files with 36 additions and 25 deletions
--- a/models.py
+++ b/models.py
@@ -241,6 +241,7 @@ class ImagBehavior(nn.Module):
            config.actor_max_std,
            config.actor_temp,
            outscale=1.0,
+            unimix_ratio=config.action_unimix_ratio,
        )  # action_dist -> action_disc?
        if config.value_head == "twohot_symlog":
            self.value = networks.DenseHead(