File size: 2,378 Bytes
9e14838 | 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 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 | datamodule:
_target_: safeear.datas.asvspoof19.DataModule
batch_size: 2
num_workers: 8
pin_memory: true
DataClass_dict:
_target_: safeear.datas.asvspoof19.DataClass
train_path: ["datas/ASVSpoof2019/train.tsv", "datas/ASVSpoof2019/ASVspoof2019.LA.cm.train.trn.txt", "datas/datasets/ASVSpoof2019_Hubert_L9/ASVspoof2019_LA_train/flac"]
val_path: ["datas/ASVSpoof2019/dev.tsv", "datas/ASVSpoof2019/ASVspoof2019.LA.cm.dev.trl.txt", "datas/datasets/ASVSpoof2019_Hubert_L9/ASVspoof2019_LA_dev/flac"]
test_path: ["datas/ASVSpoof2019/eval.tsv", "datas/ASVSpoof2019/ASVspoof2019.LA.cm.eval.trl.txt", "datas/datasets/ASVSpoof2019_Hubert_L9/ASVspoof2019_LA_eval/flac"]
max_len: 64600
decouple_model:
_target_: safeear.models.decouple.SpeechTokenizer
n_filters: 64
strides: [8,5,4,2]
dimension: 1024
semantic_dimension: 768
bidirectional: true
dilation_base: 2
residual_kernel_size: 3
n_residual_layers: 1
lstm_layers: 2
activation: ELU
codebook_size: 1024
n_q: 8
sample_rate: 16000
speechtokenizer_path: model_zoos/SpeechTokenizer.pt
detect_model:
_target_: safeear.models.safeear.SafeEar1s
front:
_target_: safeear.models.safeear.SE_Rawformer_front
embedding_dim: 1024
dropout_rate: 0.1
attention_dropout: 0.1
stochastic_depth: 0.1
num_layers: 2
num_heads: 8
num_classes: 2
positional_embedding: 'sine'
mlp_ratio: 1.0
system:
_target_: safeear.trainer.safeear_trainer.SafeEarTrainer
lr_raw_former: 3.0e-4
save_score_path: ${exp.dir}/${exp.name}
exp:
dir: Exps/ # 修改
name: ASVspoof19 # 修改
early_stopping:
_target_: pytorch_lightning.callbacks.EarlyStopping
monitor: val_eer # 修改
mode: min
patience: 40
verbose: true
checkpoint:
_target_: pytorch_lightning.callbacks.ModelCheckpoint
dirpath: ${exp.dir}/${exp.name}/checkpoints
monitor: val_eer # 修改
mode: min
verbose: true
save_top_k: 1
save_last: true
filename: '{epoch}-{val_eer:.4f}' # 修改
logger:
_target_: pytorch_lightning.loggers.WandbLogger
name: ${exp.name}
save_dir: ${exp.dir}/${exp.name}/logs
offline: true
project: SafeEar
trainer:
_target_: pytorch_lightning.Trainer
devices: [0]
max_epochs: 500
sync_batchnorm: true
default_root_dir: ${exp.dir}/${exp.name}/
accelerator: gpu
limit_train_batches: 1.0
limit_val_batches: 1.0
fast_dev_run: false |