File size: 409 Bytes
96bba34 | 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 | seed: 42
data:
processed_dir: ${oc.env:SADAR_PROCESSED_DIR,data/processed}
model:
hidden_size: 64
latent_size: 8
num_layers: 1
dropout: 0.1488
beta: 0.0001193
training:
epochs: 50
batch_size: 256
learning_rate: 0.001436
weight_decay: 0.0000054741
patience: 7
lr_factor: 0.5
lr_patience: 3
device: auto
output:
dir: ${oc.env:SADAR_MODELS_DIR,models}
checkpoint: vae_lstm.pt
|