Weights renaming (#76)

* removing deprecated train_data.py * updating weight and config filenames w version str
BitMind-AI · Oct 2, 2024 · ee7bd0c · ee7bd0c
1 parent dfa1614
commit ee7bd0c
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 293 deletions.
diff --git a/base_miner/deepfake_detectors/configs/ucf.yaml b/base_miner/deepfake_detectors/configs/ucf.yaml
@@ -1,4 +1,4 @@
 # UCFDetector Generalist Configuration
 hf_repo: 'bitmind/bm-ucf'  # Hugging Face repository for downloading model files
-train_config: 'bm-general-config.yaml'  # pre-trained configuration file in HuggingFace
-weights: 'bm-general.pth'  # UCF model checkpoint in HuggingFace
+train_config: 'bm-general-config-v1.yaml'  # pre-trained configuration file in HuggingFace
+weights: 'bm-general-v1.pth'  # UCF model checkpoint in HuggingFace
diff --git a/base_miner/deepfake_detectors/configs/ucf_face.yaml b/base_miner/deepfake_detectors/configs/ucf_face.yaml
@@ -1,4 +1,4 @@
 # UCFDetector Face Expert Configuration
 hf_repo: 'bitmind/bm-ucf'  # Hugging Face repository for downloading model files
-train_config: 'bm-faces-config.yaml'  # pre-trained configuration file in HuggingFace
-weights: 'bm-faces.pth'  # UCF model checkpoint in HuggingFace
+train_config: 'bm-faces-config-v1.yaml'  # pre-trained configuration file in HuggingFace
+weights: 'bm-faces-v1.pth'  # UCF model checkpoint in HuggingFace
diff --git a/bitmind/utils/train_data.py b/bitmind/utils/train_data.py
diff --git a/neurons/validator.py b/neurons/validator.py
@@ -56,7 +56,7 @@ def __init__(self, config=None):
 
         bt.logging.info("Loading real datasets")
         self.real_image_datasets = [
-            ImageDataset(ds['path'], 'train', ds.get('name', None), ds['create_splits'])
+            ImageDataset(ds['path'], 'train', ds.get('name', None))
             for ds in DATASET_META['real']
         ]