erased unused options

2024-01-05 23:23:09 +09:00
parent a27711ab96
commit 7f66ed5333
6 changed files with 84 additions and 211 deletions
--- a/exploration.py
+++ b/exploration.py
@@ -38,7 +38,7 @@ class Random(nn.Module):


 class Plan2Explore(nn.Module):
-    def __init__(self, config, world_model, reward=None):
+    def __init__(self, config, world_model, reward):
        super(Plan2Explore, self).__init__()
        self._config = config
        self._use_amp = True if config.precision == 16 else False