diff options
Diffstat (limited to 'configs/toy_lq.yaml')
| -rw-r--r-- | configs/toy_lq.yaml | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/configs/toy_lq.yaml b/configs/toy_lq.yaml new file mode 100644 index 0000000..ad1bba0 --- /dev/null +++ b/configs/toy_lq.yaml @@ -0,0 +1,14 @@ +d_hidden: 64 +output_dim: 10 +num_layers: 12 +sigma: 0.03 +batch_size: 256 +num_steps: 5000 +lr_fb: 0.001 +lam: 0.1 +K: 8 +ema_momentum: 0.995 +sigma_bridge: 0.03 +eval_every: 200 +gpu: 1 +output_dir: results/toy_lq |