Update train_util.py

gesen2egee · Mar 19, 2024 · 84c5ce8 · 84c5ce8
1 parent 15c8d17
commit 84c5ce8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/library/train_util.py b/library/train_util.py
@@ -1262,7 +1262,7 @@ def __getitem__(self, index):
             image_info = self.image_data[image_key]
             subset = self.image_to_subset[image_key]
             sample_weight = 1.0
-            sample_weight_path = os.path.splitext(info.absolute_path)[0] + ".weight"
+            sample_weight_path = os.path.splitext(image_info.absolute_path)[0] + ".weight"
             try:
                 with open(sample_weight_path, 'r', encoding='utf-8') as file:
                     sample_weight = float(file.readline().strip())