Spaces:
Running
on
Zero
Running
on
Zero
File size: 366 Bytes
456aee9 eebcb51 456aee9 |
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 |
in_dim: 384
mood_dim: -1
out_dim: 1280
latent_dim: 512
n_layer: 4
n_eig: 32
eigvec_loss: 1.
recon_loss_fg: 1.
recon_loss_fg_dummy: 0.0
recon_loss_bg: 0.1
recon_loss_bg_dummy: 0.0
riemann_curvature_loss: 0.0
axis_align_loss: 0.0001
repulsion_loss: 0.0001
boundary_loss: 0.1
grad_clip_val: 1.0
log_dir: /tmp/logs/
name: debug
steps: 1000
batch_size: 514
lr: 0.001
|