File size: 1,117 Bytes
010952f |
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 |
train_tsv: "<your-processed-data>/denoising/emov/train.tsv"
train_km: "<your-processed-data>/denoising/emov/train.km"
valid_tsv: "<your-processed-data>/denoising/emov/valid.tsv"
valid_km: "<your-processed-data>/denoising/emov/valid.km"
n_tokens: 200
batch_size: 32
lr: 0.0001
epochs: 300
model: "cnn"
substring: ""
rnn:
_target_: emotion_models.duration_predictor.RnnPredictor
n_tokens: ${n_tokens}
emb_dim: 128
rnn_hidden: 128
output_dim: 1
dropout: 0
n_layers: 1
optimizer:
_target_: torch.optim.Adam
lr: ${lr}
betas: [0.9, 0.98]
eps: 0.000000001
weight_decay: 0
cnn:
_target_: emotion_models.duration_predictor.CnnPredictor
n_tokens: ${n_tokens}
emb_dim: 128
channels: 256
kernel: 3
output_dim: 1
dropout: 0.5
n_layers: 1
hydra:
run:
dir: /checkpoint/felixkreuk/experiments/duration_predictor/${hydra.job.override_dirname}
job:
config:
# configuration for the ${hydra.job.override_dirname} runtime variable
override_dirname:
kv_sep: '='
item_sep: ','
exclude_keys: ['train_tsv', 'train_km', 'valid_tsv', 'valid_km']
|