Skip to content

Commit

Permalink
batch conf
Browse files Browse the repository at this point in the history
  • Loading branch information
ihounie committed Mar 15, 2023
1 parent 172f192 commit ef32c25
Showing 1 changed file with 38 additions and 0 deletions.
38 changes: 38 additions & 0 deletions confs/ablations/batch/pd_cifar100_ablation.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,38 @@
model:
type: wresnet40_2
dataset: cifar100
aug: primaldual
randaug:
N: 0
M: 0 # from appendix
weights: [0.0,1.0]
cutout: 16
batch: 128
gpus: 1
n_aug_set:
- 1
- 2
- 4
- 8
- 16
augmentation_search_space: wide_standard
epoch: 200
lr: 0.1
lr_schedule:
type: 'cosine'
optimizer:
type: sgd
nesterov: True
decay: 0.0005
MH:
steps: 2
PD:
lr: 0.0001
margin: 2.1
sample: False
save_model: True
seed_set:
- 0
- 1
- 2
- 3

0 comments on commit ef32c25

Please sign in to comment.