fusion-bench 0.2.7__py3-none-any.whl → 0.2.8__py3-none-any.whl

fusion_bench/tasks/clip_classification/food101.py

@@ -0,0 +1,105 @@
+classnames = [
+    "apple pie",
+    "baby back ribs",
+    "baklava",
+    "beef carpaccio",
+    "beef tartare",
+    "beet salad",
+    "beignets",
+    "bibimbap",
+    "bread pudding",
+    "breakfast burrito",
+    "bruschetta",
+    "caesar salad",
+    "cannoli",
+    "caprese salad",
+    "carrot cake",
+    "ceviche",
+    "cheese plate",
+    "cheesecake",
+    "chicken curry",
+    "chicken quesadilla",
+    "chicken wings",
+    "chocolate cake",
+    "chocolate mousse",
+    "churros",
+    "clam chowder",
+    "club sandwich",
+    "crab cakes",
+    "creme brulee",
+    "croque madame",
+    "cup cakes",
+    "deviled eggs",
+    "donuts",
+    "dumplings",
+    "edamame",
+    "eggs benedict",
+    "escargots",
+    "falafel",
+    "filet mignon",
+    "fish and chips",
+    "foie gras",
+    "french fries",
+    "french onion soup",
+    "french toast",
+    "fried calamari",
+    "fried rice",
+    "frozen yogurt",
+    "garlic bread",
+    "gnocchi",
+    "greek salad",
+    "grilled cheese sandwich",
+    "grilled salmon",
+    "guacamole",
+    "gyoza",
+    "hamburger",
+    "hot and sour soup",
+    "hot dog",
+    "huevos rancheros",
+    "hummus",
+    "ice cream",
+    "lasagna",
+    "lobster bisque",
+    "lobster roll sandwich",
+    "macaroni and cheese",
+    "macarons",
+    "miso soup",
+    "mussels",
+    "nachos",
+    "omelette",
+    "onion rings",
+    "oysters",
+    "pad thai",
+    "paella",
+    "pancakes",
+    "panna cotta",
+    "peking duck",
+    "pho",
+    "pizza",
+    "pork chop",
+    "poutine",
+    "prime rib",
+    "pulled pork sandwich",
+    "ramen",
+    "ravioli",
+    "red velvet cake",
+    "risotto",
+    "samosa",
+    "sashimi",
+    "scallops",
+    "seaweed salad",
+    "shrimp and grits",
+    "spaghetti bolognese",
+    "spaghetti carbonara",
+    "spring rolls",
+    "steak",
+    "strawberry shortcake",
+    "sushi",
+    "tacos",
+    "takoyaki",
+    "tiramisu",
+    "tuna tartare",
+    "waffles",
+]
+
+templates = [lambda c: f"a photo of {c}, a type of food."]

fusion_bench/tasks/clip_classification/kmnist.py

@@ -0,0 +1,17 @@
+classnames_mapping = {
+    "0": "お",
+    "1": "き",
+    "2": "す",
+    "3": "つ",
+    "4": "な",
+    "5": "は",
+    "6": "ま",
+    "7": "や",
+    "8": "れ",
+    "9": "を",
+}
+classnames = [classnames_mapping[str(c)] for c in range(10)]
+
+templates = [
+    lambda c: f"a photo of the character {c}.",
+]

fusion_bench/tasks/clip_classification/mongo_leaf_disease.py

@@ -0,0 +1,19 @@
+classnames = [
+    "Anthracnose",
+    "Bacterial Canker",
+    "Cutting Weevil",
+    "Die Back",
+    "Gall Midge",
+    "Healthy",
+    "Powdery Mildew",
+    "Sooty Mould",
+]
+
+templates = [
+    lambda c: f"a photo of a mango leaf with {c}.",
+    lambda c: f"a mango leaf showing symptoms of {c}.",
+    lambda c: f"a close-up photo of {c} on a mango leaf.",
+    lambda c: f"this mango leaf is affected by {c}.",
+    lambda c: f"a mango leaf disease identified as {c}.",
+    lambda c: f"a {c} infection on a mango leaf.",
+]

fusion_bench/tasks/clip_classification/pcam.py

@@ -0,0 +1,5 @@
+classnames = ["lymph node", "lymph node containing metastatic tumor tissue"]
+
+templates = [
+    lambda c: f"this is a photo of {c}",
+]

fusion_bench/utils/parameters.py

@@ -1,6 +1,6 @@
 import copy
 from collections import OrderedDict
-from typing import List, Mapping, Union
+from typing import List, Mapping, Optional, Union
 
 import torch
 from torch import nn
@@ -43,7 +43,10 @@ def trainable_state_dict(
     return state_dict
 
 
-def state_dict_to_vector(state_dict, remove_keys=[]):
+def state_dict_to_vector(
+    state_dict: StateDictType,
+    remove_keys: Optional[List[str]] = None,
+):
     """
     Convert a state dictionary to a vector.
 
@@ -54,6 +57,7 @@ def state_dict_to_vector(state_dict, remove_keys=[]):
     Returns:
         torch.Tensor: The converted vector.
     """
+    remove_keys = remove_keys if remove_keys is not None else []
     shared_state_dict = copy.deepcopy(state_dict)
     for key in remove_keys:
         if key in shared_state_dict:
@@ -64,7 +68,11 @@ def state_dict_to_vector(state_dict, remove_keys=[]):
     )
 
 
-def vector_to_state_dict(vector, state_dict, remove_keys=[]):
+def vector_to_state_dict(
+    vector: torch.Tensor,
+    state_dict: StateDictType,
+    remove_keys: Optional[List[str]] = None,
+):
     """
     Convert a vector to a state dictionary.
 
@@ -76,6 +84,7 @@ def vector_to_state_dict(vector, state_dict, remove_keys=[]):
     Returns:
         dict: The converted state dictionary.
     """
+    remove_keys = remove_keys if remove_keys is not None else []
     # create a reference dict to define the order of the vector
     reference_dict = copy.deepcopy(state_dict)
     for key in remove_keys:

fusion_bench/utils/type.py

@@ -22,4 +22,13 @@ T2 = TypeVar("T2")
 T3 = TypeVar("T3")
 T4 = TypeVar("T4")
 
-__all__ = ["StateDictType", "PyModuleType", "TorchModelType", "T", "T1", "T2", "T3", "T4"]
+__all__ = [
+    "StateDictType",
+    "PyModuleType",
+    "TorchModelType",
+    "T",
+    "T1",
+    "T2",
+    "T3",
+    "T4",
+]

{fusion_bench-0.2.7.dist-info → fusion_bench-0.2.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fusion_bench
-Version: 0.2.7
+Version: 0.2.8
 Summary: A Comprehensive Benchmark of Deep Model Fusion
 Author-email: Anke Tang <tang.anke@foxmail.com>
 License: MIT License