added Docker instruction

2023-10-02 07:37:08 +09:00
parent 1002d8b115
commit 0eec12bd1c
4 changed files with 16 additions and 10 deletions
--- a/exploration.py
+++ b/exploration.py
@@ -58,9 +58,10 @@ class Plan2Explore(nn.Module):
            "feat": config.dyn_stoch + config.dyn_deter,
        }[self._config.disag_target]
        kw = dict(
-            inp_dim=feat_size + (config.num_actions
-            if config.disag_action_cond
-            else 0),  # pytorch version
+            inp_dim=feat_size
+            + (
+                config.num_actions if config.disag_action_cond else 0
+            ),  # pytorch version
            shape=size,
            layers=config.disag_layers,
            units=config.disag_units,