Skip to content

Instantly share code, notes, and snippets.

@juntang-zhuang
Created April 5, 2021 19:19
Show Gist options
  • Star 1 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save juntang-zhuang/517ce3c27022b908bb93f78e4f786dc3 to your computer and use it in GitHub Desktop.
Save juntang-zhuang/517ce3c27022b908bb93f78e4f786dc3 to your computer and use it in GitHub Desktop.
args.yaml file to train efficientnet-b0 with adabelief
aa: rand-m9-mstd0.5
amp: true
apex_amp: false
aug_splits: 0
batch_size: 192
bn_eps: null
bn_momentum: 0.1
bn_tf: false
channels_last: false
checkpoint_hist: 10
clip_grad: null
clip_mode: norm
color_jitter: 0.4
cooldown_epochs: 10
crop_pct: null
cutmix: 0.0
cutmix_minmax: null
data_dir: /media/juntang/Samsung_T5/ImageNet
dataset: ''
decay_epochs: 30
decay_rate: 0.1
dist_bn: ''
drop: 0.2
drop_block: null
drop_connect: null
drop_path: 0.2
epoch_repeats: 0.0
epochs: 400
eval_metric: top1
experiment: ''
gp: null
hflip: 0.5
img_size: null
initial_checkpoint: ''
input_size: null
interpolation: ''
jsd: false
local_rank: 0
log_interval: 50
lr: 0.002
lr_cycle_limit: 1
lr_cycle_mul: 1.0
lr_noise: null
lr_noise_pct: 0.67
lr_noise_std: 1.0
mean: null
min_lr: 1.0e-05
mixup: 0.2
mixup_mode: batch
mixup_off_epoch: 400
mixup_prob: 1.0
mixup_switch_prob: 0.5
model: efficientnet_b0
model_ema: false
model_ema_decay: 0.9998
model_ema_force_cpu: false
momentum: 0.9
native_amp: false
no_aug: false
no_prefetcher: false
no_resume_opt: false
num_classes: null
opt: adabelief
opt_betas: null
opt_eps: null
output: ''
patience_epochs: 10
pin_mem: false
pretrained: false
ratio:
- 0.75
- 1.3333333333333333
recount: 1
recovery_interval: 0
remode: pixel
reprob: 0.2
resplit: false
resume: ''
save_images: false
scale:
- 0.08
- 1.0
sched: cosine
seed: 42
smoothing: 0.1
split_bn: false
start_epoch: null
std: null
sync_bn: false
torchscript: false
train_interpolation: random
train_split: train
tta: 0
use_multi_epochs_loader: false
val_split: validation
validation_batch_size_multiplier: 1
vflip: 0.0
warmup_epochs: 3
warmup_lr: 0.0001
weight_decay: 0.025
workers: 8
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment