From 32703efe7412faa88d1883113eead11068d3db5d Mon Sep 17 00:00:00 2001 From: Niko Date: Thu, 24 Aug 2023 14:23:41 +0200 Subject: [PATCH] Deepmind suite tested --- PolicyModel/GaussianModelMultiDim.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PolicyModel/GaussianModelMultiDim.py b/PolicyModel/GaussianModelMultiDim.py index 76c2ff0..5d9030c 100644 --- 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, )