Update LFNA version 1.0

2021-05-13 21:33:34 +08:00
parent 3d3a04705f
commit cfabd05de8
11 changed files with 340 additions and 299 deletions
--- a/lib/datasets/synthetic_utils.py
+++ b/lib/datasets/synthetic_utils.py
@@ -30,6 +30,7 @@ class UnifiedSplit:
            self._indexes = all_indexes[num_of_train + num_of_valid :]
        else:
            raise ValueError("Unkonwn mode of {:}".format(mode))
+        self._all_indexes = all_indexes
        self._mode = mode

    @property