Skip to content

fusion_bench.constants

Paths

fusion_bench.constants.paths

PROJECT_ROOT_PATH = LIBRARY_PATH.parent module-attribute

Path to the project root directory.

LIBRARY_PATH = Path(importlib.import_module('fusion_bench').__path__[0]) module-attribute

Path to the library directory.

DEFAULT_CONFIG_PATH = PROJECT_ROOT_PATH / 'config' module-attribute

Path to the default config directory.

CLIP Vision Tasks

fusion_bench.constants.clip_vision

Constants for CLIP Vision Model Merging

TASK_NAMES_TA8 = ['sun397', 'stanford-cars', 'resisc45', 'eurosat', 'svhn', 'gtsrb', 'mnist', 'dtd'] module-attribute

The 8 tasks used in the Task Arithmetic paper.

TASK_NAMES_TALL8 = TASK_NAMES_TA8 module-attribute

The 8 tasks used in the Tall Mask paper

TASK_NAMES_TALL10 = TASK_NAMES_TA8 + ['oxford_flowers102', 'pcam'] module-attribute

TASK_NAMES_TALL12 = TASK_NAMES_TALL10 + ['fer2013', 'oxford-iiit-pet'] module-attribute

TASK_NAMES_TALL14 = TASK_NAMES_TALL12 + ['stl10', 'cifar100'] module-attribute

The 14 tasks used in the TALL mask paper

TASK_NAMES_TALL16 = TASK_NAMES_TALL14 + ['cifar10', 'food101'] module-attribute

TASK_NAMES_TALL18 = TASK_NAMES_TALL16 + ['fashion_mnist', 'emnist_letters'] module-attribute

TASK_NAMES_TALL20 = TASK_NAMES_TALL18 + ['kmnist', 'rendered-sst2'] module-attribute

The 20 tasks used in the TALL mask paper

TASK_NAMES_TA8_CAP = ['SUN397', 'Cars', 'RESISC45', 'EuroSAT', 'SVHN', 'GTSRB', 'MNIST', 'DTD'] module-attribute

TASK_NAMES_TALL8_CAP = TASK_NAMES_TA8_CAP module-attribute

TASK_NAMES_TALL10_CAP = TASK_NAMES_TALL8_CAP + ['Flowers102', 'PCAM'] module-attribute

TASK_NAMES_TALL12_CAP = TASK_NAMES_TALL10_CAP + ['FER2013', 'OxfordIIITPet'] module-attribute

TASK_NAMES_TALL14_CAP = TASK_NAMES_TALL12_CAP + ['STL10', 'CIFAR100'] module-attribute

TASK_NAMES_TALL16_CAP = TASK_NAMES_TALL14_CAP + ['CIFAR10', 'Food101'] module-attribute

TASK_NAMES_TALL18_CAP = TASK_NAMES_TALL16_CAP + ['FashionMNIST', 'EMNIST'] module-attribute

TASK_NAMES_TALL20_CAP = TASK_NAMES_TALL18_CAP + ['KMNIST', 'RenderedSST2'] module-attribute