fix(pu): fix lstm_hidden_size in sampled_efficientzero_model.py

opendilab · Dec 21, 2023 · 3823560 · 3823560
1 parent 1958861
commit 3823560
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lzero/model/sampled_efficientzero_model.py b/lzero/model/sampled_efficientzero_model.py
@@ -182,6 +182,7 @@ def __init__(
             self.reward_support_size,
             flatten_output_size_for_reward_head,
             downsample,
+            lstm_hidden_size=self.lstm_hidden_size,
             last_linear_layer_init_zero=self.last_linear_layer_init_zero,
             activation=activation,
             norm_type=norm_type