Yixuan's picture
Upload folder using huggingface_hub
50dfee0 verified
criterion:
feat_dim: 256
model_class: OCSoftmax
fake_weight: 1.0
real_weight: 9.0
data:
augment_type: rawboost
batch_size: 128
data_path: dataset_meta/ITW_train
do_augment: true
max_samples: 64000
model_class: GeneralFAD
num_workers: 12
test_batch_size: 64
test_num_workers: 4
upsample: false
model:
cuda: 0
frontend: XLSR
frontend_path: /mount/studenten/arbeitsdaten-studenten3/advanced-ml/2022/YixuanXiao/workenv/fad/download/model/xlsr2_300m.pt
model_class: XLSRAdapter
num_classes: 1
train:
beta1: 0.9
beta2: 0.999
log_interval: 20
lr: 0.0001
lr_gamma: 0.5
lr_step_size: 20
max_epoch: 100
max_steps: 10000000000
start_steps: 0
validate_interval: 10000000000
weight_decay: 0.0001
patience: 10