Fix bugs

2021-05-25 20:24:41 -07:00
parent 9057011781
commit 50f4f3dda7
3 changed files with 5 additions and 2 deletions
--- a/xautodl/datasets/math_base_funcs.py
+++ b/xautodl/datasets/math_base_funcs.py
@@ -28,7 +28,9 @@ class FitFunc(abc.ABC):
        self._params = copy.deepcopy(params)

    def check_valid(self):
-        for key, value in self._params.items():
+        # for key, value in self._params.items():
+        for key in range(self._freedom):
+            value = self._params[key]
            if value is None:
                raise ValueError("The {:} is None".format(key))