| 30 | |
| 31 | |
| 32 | class HyperParamsStrandVAE: |
| 33 | def __init__(self): |
| 34 | self.nr_verts_per_strand=256 |
| 35 | self.use_fourier_space_strands=False |
| 36 | self.enable_vae = True |
| 37 | |
| 38 | self.nr_iters_to_train=3000000 |
| 39 | self.lr= 3e-3 #for batch size 10 |
| 40 | self.save_checkpoint=True |
| 41 | self.save_checkpoint_every_x_epoch=10 |
| 42 | self.with_tensorboard=True |
| 43 | self.with_visualizer=False |
| 44 | self.viewer_config_path=os.path.join(SCRIPT_DIR, "./configs/strand_vae_train.toml") |
| 45 | |
| 46 | #####Input |
| 47 | self.normalize_input=True |
| 48 | |
| 49 | |
| 50 | #####Output |
| 51 | self.decode_type="dir" |
| 52 | self.scale_init=30.0 |
| 53 | self.nr_verts_per_strand=256 |
| 54 | self.nr_values_to_decode=255 |
| 55 | self.dim_per_value_decoded=3 |
| 56 | |
| 57 | |
| 58 | ###LOSS###### |
| 59 | self.loss_pos_weight=0.5 |
| 60 | self.loss_dir_weight=1.0 |
| 61 | self.loss_curv_weight=20.0 |
| 62 | self.loss_kl_weight=6e-4 |
| 63 | |
| 64 | |
| 65 | |