diff --git a/base_miner/deepfake_detectors/configs/ucf.yaml b/base_miner/deepfake_detectors/configs/ucf.yaml index 9c7ec71..a8ae766 100644 --- 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 \ No newline at end of file +train_config: 'bm-general-config-v1.yaml' # pre-trained configuration file in HuggingFace +weights: 'bm-general-v1.pth' # UCF model checkpoint in HuggingFace \ No newline at end of file diff --git a/base_miner/deepfake_detectors/configs/ucf_face.yaml b/base_miner/deepfake_detectors/configs/ucf_face.yaml index 0f96702..9de7721 100644 --- 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/neurons/validator.py b/neurons/validator.py index d8ad6e9..19dab1e 100644 --- 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'] ]