Merge pull request DeNA#280 from YuriCat/fix/remove_hidden_output_fro…

…m_random_model2 fix: remove hidden output from random model
YuriCat · Feb 25, 2022 · d131669 · d131669
2 parents 627def8 + 00ec537
commit d131669
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/handyrl/model.py b/handyrl/model.py
@@ -68,7 +68,7 @@ def __init__(self, model, x):
         wrapped_model = ModelWrapper(model)
         hidden = wrapped_model.init_hidden()
         outputs = wrapped_model.inference(x, hidden)
-        self.output_dict = {key: np.zeros_like(value) for key, value in outputs.items()}
+        self.output_dict = {key: np.zeros_like(value) for key, value in outputs.items() if key != 'hidden'}
 
     def inference(self, *args):
         return self.output_dict