Re-organize GeMOSA

2021-05-27 15:56:08 +08:00
parent 6da60664f5
commit 4ca38ad2ff
6 changed files with 26 additions and 19 deletions
--- a/xautodl/datasets/math_dynamic_generator.py
+++ b/xautodl/datasets/math_dynamic_generator.py
@@ -47,6 +47,9 @@ class GaussianDGenerator(DynamicGenerator):
    def ndim(self):
        return self._ndim

+    def output_shape(self):
+        return (self._ndim,)
+
    def __call__(self, time, num):
        mean_list = [functor(time) for functor in self._mean_functors]
        cov_matrix = [