modifications for minecraft

2023-08-05 21:13:57 +09:00
parent 8c471e12d6
commit 8571cf656a
3 changed files with 21 additions and 16 deletions
--- a/networks.py
+++ b/networks.py
@@ -415,7 +415,7 @@ class MultiDecoder(nn.Module):
        vector_dist,
    ):
        super(MultiDecoder, self).__init__()
-        excluded = ("is_first", "is_last", "is_terminal", "reward")
+        excluded = ("is_first", "is_last", "is_terminal")
        shapes = {k: v for k, v in shapes.items() if k not in excluded}
        self.cnn_shapes = {
            k: v for k, v in shapes.items() if len(v) == 3 and re.match(cnn_keys, k)