merged action head into MLP and modified configs

2024-01-05 10:26:48 +09:00
parent e0f2017e28
commit e0487f8206
5 changed files with 133 additions and 231 deletions
--- a/exploration.py
+++ b/exploration.py
@@ -14,7 +14,7 @@ class Random(nn.Module):
        self._act_space = act_space

    def actor(self, feat):
-        if self._config.actor_dist == "onehot":
+        if self._config.actor["dist"] == "onehot":
            return tools.OneHotDist(
                torch.zeros(self._config.num_actions)
                .repeat(self._config.envs, 1)