Update GeMOSA v4

2021-05-27 17:30:44 +08:00
parent 1ce0b80776
commit b6e11c6360
8 changed files with 147 additions and 39 deletions
--- a/xautodl/datasets/math_core.py
+++ b/xautodl/datasets/math_core.py
@@ -10,5 +10,10 @@ from .math_static_funcs import (
    ComposedSinSFunc,
    ComposedCosSFunc,
 )
-from .math_dynamic_funcs import LinearDFunc, QuadraticDFunc, SinQuadraticDFunc
-from .math_dynamic_generator import GaussianDGenerator
+from .math_dynamic_funcs import (
+    LinearDFunc,
+    QuadraticDFunc,
+    SinQuadraticDFunc,
+    BinaryQuadraticDFunc,
+)
+from .math_dynamic_generator import UniformDGenerator, GaussianDGenerator