fusion-bench 0.2.28__py3-none-any.whl → 0.2.30__py3-none-any.whl

fusion_bench_config/llama_full_finetune.yaml

@@ -1,18 +1,6 @@
 defaults:
-  - hydra: default
-  - fabric: llama_fsdp
-  - path: default
-  # --- Model, Method, Task ---
-  - method: lm_finetune/fullfinetune_sft.yaml
-  - modelpool: CausalLMPool/llama_alpaca_cleaned.yaml
-  - taskpool: dummy
+  - fabric_model_fusion
+  - override fabric: llama_fsdp
+  - override method: lm_finetune/fullfinetune_sft.yaml
+  - override modelpool: CausalLMPool/llama_alpaca_cleaned.yaml
   - _self_
-_target_: fusion_bench.programs.FabricModelFusionProgram
-_recursive_: false
-fast_dev_run: false # Run a single batch of data to test the model or method
-# Run the script without actually running the experiment, use with `print_config=true`.
-# You can also use `--cfg` or `-c` to show the configuration instead of running.
-dry_run: false
-print_config: true # Print the configuration to the console
-report_save_path: null # path to save the result report
-print_function_call: true # set to false if you don't want to print the details of instantiate calls

fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_robustness_corrupted.yaml

@@ -6,7 +6,7 @@ defaults:
       - clip-vit-base-patch32_eurosat
       - clip-vit-base-patch32_resisc45
       - clip-vit-base-patch32_gtsrb
-# `corrption` can be one of:
+# `corruption` can be one of:
 # contrast, gaussian_noise, impulse_noise, jpeg_compression, motion_blur, pixelate, spatter
 corruption: ${corruption}
 # The following datasets are used for test-time adaptation

fusion_bench_config/nyuv2_config.yaml

@@ -1,17 +1,8 @@
 defaults:
-  - hydra: default
-  - fabric: auto
-  - path: default
-  # --- Model, Method, Task ---
-  - method: simple_average
-  - modelpool: nyuv2_modelpool
-  - taskpool: nyuv2_taskpool
+  - fabric_model_fusion
+  - override method: simple_average
+  - override modelpool: nyuv2_modelpool
+  - override taskpool: nyuv2_taskpool
   - _self_
-_target_: fusion_bench.programs.FabricModelFusionProgram
-_recursive_: false
-fast_dev_run: false # Run a single batch of data to test the model or method
-use_lightning: true # Use the fabric to run the experiment
-print_config: true # Print the configuration to the console
-save_report: false # path to save the result report
 trainer:
   devices: 1

fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-base-patch32_robustness_corrupted.yaml

@@ -1,6 +1,6 @@
 type: clip_vit_classification
 name: clip-vit-robustness_clean
-# corrption can be one of:
+# corruption can be one of:
 # contrast, gaussian_noise, impulse_noise, jpeg_compression, motion_blur, pixelate, spatter
 corruption: ${corruption}
 dataset_type: huggingface_image_classification

fusion_bench_config/taskpool/clip-vit-base-patch32_robustness_corrupted.yaml

@@ -1,6 +1,6 @@
 type: clip_vit_classification
 name: clip-vit-robustness_clean
-# corrption can be one of:
+# corruption can be one of:
 # contrast, gaussian_noise, impulse_noise, jpeg_compression, motion_blur, pixelate, spatter
 corruption: ${corruption}
 dataset_type: huggingface_image_classification

fusion_bench/utils/auto.py

@@ -1,31 +0,0 @@
-from omegaconf import DictConfig
-
-from fusion_bench.utils import import_object
-
-
-class BaseFactoryClass:
-    _registry = {}
-
-    @classmethod
-    def from_config(cls, config: DictConfig):
-        name = config.name
-        if name not in cls._registry:
-            raise ValueError(
-                f"Unknown name: {name}, available names: {cls._registry.keys()}. "
-                f"You can register a new item using `{cls.__name__}.register()` method."
-            )
-
-        item_cls = cls._registry[name]
-        if isinstance(item_cls, str):
-            if item_cls.startswith("."):
-                item_cls = f"{cls.__module__}.{item_cls[1:]}"
-            item_cls = import_object(item_cls)
-        return item_cls(config)
-
-    @classmethod
-    def register(cls, name: str, item_cls):
-        cls._registry[name] = item_cls
-
-    @classmethod
-    def available_items(cls):
-        return list(cls._registry.keys())