| 54 | |
| 55 | |
| 56 | class HyperParamsStrandVAE: |
| 57 | def __init__(self): |
| 58 | #####Output |
| 59 | #####decode dir###### |
| 60 | self.decode_type="dir" |
| 61 | self.scale_init=30.0 |
| 62 | self.nr_verts_per_strand=256 |
| 63 | self.nr_values_to_decode=255 |
| 64 | self.dim_per_value_decoded=3 |
| 65 | |
| 66 | |
| 67 | ###LOSS###### |
| 68 | #these are the same values that were used to train the strand vae |
| 69 | self.loss_pos_weight=0.5 ##FOR LAMBDA |
| 70 | self.loss_dir_weight=1.0 |
| 71 | self.loss_curv_weight=20.0 |
| 72 | self.loss_kl_weight=0.0 |
| 73 | |
| 74 | |
| 75 |