File size: 2,480 Bytes
8617847 | 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 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 | checkpoint: exp/redimnet2/expB6_v2/-LM/models/model_0.pt
data_type: shard
dataloader_args:
batch_size: 16
drop_last: true
num_workers: 4
pin_memory: false
prefetch_factor: 4
dataset_args:
aug_prob: 0.0
cmvn: false
filter: false
frontend: tfmel
num_frms: 600
resample_rate: 16000
sample_num_per_epoch: 0
shuffle: true
shuffle_args:
shuffle_size: 2500
spec_aug: false
speed_perturb: false
tfmel_args:
do_preemph: true
do_spec_aug: false
eps: 1.0e-08
f_max: 7600
f_min: 20
hop_length: 160
n_fft: 512
n_mels: 72
norm_signal: true
win_length: 400
enable_amp: true
exp_dir: exp/redimnet2/expB6_v2/-LM
gpus:
- 2
- 3
grad_accum_steps: 8
log_batch_interval: 100
loss: CrossEntropyLoss
loss_args: {}
margin_scheduler: MarginScheduler
margin_update:
epoch_iter: 34125
final_margin: 0.3
fix_start_epoch: 0
increase_start_epoch: 0
increase_type: exp
initial_margin: 0.3
update_margin: false
model: ReDimNet2Wrap
model_args:
C: 64
F: 72
block_1d_type: conv+att
block_2d_type: basic_resnet
causal: none
compress_tconvs: true
emb_bn: false
embed_dim: 192
feat_dim: 72
fm_weigthing_type: NC
global_context_att: true
group_divisor: 1
hop_length: 160
out_channels: 224
pooling_func: ASTP
return_2d_output: true
spec: null
spec_in_channels: 1
stages_setup:
- - - 1
- 1
- 3
- 3
- - - 3
- 3
- 64
- - - 2
- 1
- 4
- 2
- - - 3
- 3
- 64
- - - 1
- 2
- 5
- 2
- - - 3
- 3
- 48
- - - 2
- 1
- 5
- 1
- - - 3
- 3
- 48
- - - 1
- 2
- 4
- 0.75
- - - 3
- 3
- 32
- - - 2
- 1
- 3
- 0.5
- - - 3
- 3
- 24
model_init: null
noise_data: data/voxceleb/musan/lmdb
num_avg: 1
num_epochs: 5
optimizer: SGD
optimizer_args:
lr: 0.0001
momentum: 0.9
nesterov: true
weight_decay: 2.0e-05
projection_args:
do_lm: false
easy_margin: false
embed_dim: 192
lanbuda: 0.7
margin_type: C
num_class: 5994
project_type: sphereface2
scale: 32.0
t: 3
reverb_data: data/voxceleb/rirs/lmdb
save_epoch_interval: 1
scheduler: ExponentialDecrease
scheduler_args:
epoch_iter: 34125
final_lr: 1.0e-05
initial_lr: 0.0001
num_epochs: 5
scale_ratio: 4.0
warm_from_zero: false
warm_up_epoch: 0
seed: 42
train_data: data/voxceleb/vox2_dev/shard.list
train_label: data/voxceleb/vox2_dev/utt2spk
|