File size: 1,593 Bytes
3b11729 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 | 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/eurosat/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: eurosat
train_datasets:
eurosat:
_target_: datasets.load_dataset
path: tanganke/eurosat
split: train
val_datasets:
eurosat:
_target_: datasets.load_dataset
path: tanganke/eurosat
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'
|