PyTorch
ssl-aasist
custom_code
ash56's picture
Add files using upload-large-folder tool
010952f verified
raw
history blame
1.12 kB
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']