File size: 491 Bytes
1c2a363 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
defaults:
- _autrainer_
- _self_
# - override /hydra/launcher: submitit_slurm
results_dir: results
experiment_id: default
iterations: 30
inference_batch_size: 1
progress_bar: false
hydra:
sweeper:
params:
+seed: 1
+batch_size: 32
+learning_rate: 0.001 #, 0.0001
dataset: InsectSetDSBalanced-96k
model: Cnn10-32k-T #, Cnn14-32k-T, ResNet-50-T, Cnn10-32k-T
optimizer: Adam
augmentation: CustomAugPipeline_woAddSpec # SpecAugment, None
|