Fix bugs in new env-v1

2021-05-24 05:14:39 +00:00
parent 3ee0d348af
commit 3a2af8e55a
4 changed files with 26 additions and 38 deletions
--- a/xautodl/datasets/synthetic_env.py
+++ b/xautodl/datasets/synthetic_env.py
@@ -115,7 +115,7 @@ class SyntheticDEnv(data.Dataset):
            name=self.__class__.__name__,
            cur_num=len(self),
            total=len(self._time_generator),
-            ndim=self._ndim,
+            ndim=self._data_generator.ndim,
            num_per_task=self._num_per_task,
            xrange_min=self.min_timestamp,
            xrange_max=self.max_timestamp,