path:
  root_dir: ${oc.env:FUSION_BENCH_PROJECT_ROOT,"."}
  output_dir: ${.root_dir}/outputs
  data_dir: ${oc.env:FUSION_BENCH_DATA_DIR,${.root_dir}/data}
  cache_dir: ${oc.env:FUSION_BENCH_CACHE_DIR,${.output_dir}/cache}
  log_dir: outputs/convnext-base-224/gtsrb/batch_size=64,lr=0.01
  work_dir: ${hydra:runtime.cwd}
modelpool:
  _target_: fusion_bench.modelpool.ConvNextForImageClassificationPool
  _recursive_: false
  models:
    _pretrained_:
      config_path: facebook/convnext-base-224
      pretrained: true
      dataset_name: gtsrb
  train_datasets:
    gtsrb:
      _target_: datasets.load_dataset
      path: tanganke/gtsrb
      split: train
  val_datasets:
    gtsrb:
      _target_: datasets.load_dataset
      path: tanganke/gtsrb
      split: test
  test_datasets: null
method:
  _target_: fusion_bench.method.classification.ImageClassificationFineTuning
  max_epochs: -1
  max_steps: 4000
  save_top_k: -1
  save_interval: 1000
  save_on_train_epoch_end: false
  training_data_ratio: null
  label_smoothing: 0
  optimizer:
    _target_: torch.optim.SGD
    lr: 0.01
    momentum: 0.9
    weight_decay: 0.0001
  lr_scheduler: null
  dataloader_kwargs:
    batch_size: 64
    num_workers: 8
    pin_memory: true
taskpool:
  _target_: fusion_bench.taskpool.DummyTaskPool
  model_save_path: null
_target_: fusion_bench.programs.ModelFusionProgram
_recursive_: false
seed: 0
fast_dev_run: false
dry_run: false
print_config: true
print_function_call: true
merged_model_save_path: null
merged_model_save_kwargs: null
report_save_path: '{log_dir}/program_report.json'