added state input capability

2023-05-14 23:38:46 +09:00
parent 3ebb8ad617
commit b984e69b6e
8 changed files with 369 additions and 142 deletions
--- a/exploration.py
+++ b/exploration.py
@@ -52,7 +52,7 @@ class Plan2Explore(nn.Module):
            act=config.act,
        )
        self._networks = nn.ModuleList(
-            [networks.DenseHead(**kw) for _ in range(config.disag_models)]
+            [networks.MLP(**kw) for _ in range(config.disag_models)]
        )
        kw = dict(wd=config.weight_decay, opt=config.opt, use_amp=self._use_amp)
        self._model_opt = tools.Optimizer(