import mix_entropy

deepmodeling · Dec 19, 2024 · 15af914 · 15af914
1 parent 124f432
commit 15af914
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/deepmd/pt/utils/dataloader.py b/deepmd/pt/utils/dataloader.py
@@ -36,6 +36,7 @@
 from deepmd.pt.utils.dataset import (
     DeepmdDataSetForLoader,
 )
+from deepmd.pt.utils.utils import mix_entropy
 from deepmd.utils.data import (
     DataRequirementItem,
 )