_recursive_: false
_target_: fusion_bench.method.classification.image_classification_finetune.ImageClassificationFineTuning
_usage_: null
_version_: 0.2.31.dev0
dataloader_kwargs:
  batch_size: 64
  num_workers: 8
  pin_memory: true
label_smoothing: 0
lr_scheduler: null
max_epochs: -1
max_steps: 4000
optimizer:
  _target_: torch.optim.SGD
  lr: 0.01
  momentum: 0.9
  weight_decay: 0.0001
save_interval: 1000
save_on_train_epoch_end: false
save_top_k: -1
training_data_ratio: null

Model Pool Configuration

yaml

_recursive_: false
_target_: fusion_bench.modelpool.convnext_for_image_classification.ConvNextForImageClassificationPool
_usage_: null
_version_: 0.2.31.dev0
models:
  _pretrained_:
    config_path: facebook/convnext-base-224
    dataset_name: emnist_letters
    pretrained: true
test_datasets: null
train_datasets:
  emnist_letters:
    _target_: datasets.load_dataset
    path: tanganke/emnist_letters
    split: train
val_datasets:
  emnist_letters:
    _target_: datasets.load_dataset
    path: tanganke/emnist_letters
    split: test

tanganke/convnext-base-224_emnist_letters_sgd_batch-size-64_lr-0.01_steps-4000

Deep Model Fusion

Models Merged

Configuration

Algorithm Configuration

Model Pool Configuration

tanganke/
convnext-base-224_emnist_letters_sgd_batch-size-64_lr-0.01_steps-4000