Deepmind suite tested

2023-08-24 14:23:41 +02:00 · 2023-08-24 14:23:41 +02:00 · 32703efe74
commit 32703efe74
parent e574a5ca2c
1 changed files with 1 additions and 1 deletions
--- a/PolicyModel/GaussianModelMultiDim.py
+++ b/PolicyModel/GaussianModelMultiDim.py
@ -46,4 +46,4 @@ class GaussianPolicy:
        self.weights = x.reshape(self.nr_weights, self.nr_dims)

    def get_x(self):
-        return self.weights.reshape(self.nr_weights * self.nr_dims, 1)
+        return self.weights.reshape(self.nr_weights * self.nr_dims, )