| 32 | |
| 33 | |
| 34 | class HyperParamsRGB2Mat: |
| 35 | def __init__(self): |
| 36 | self.nr_iters_to_train=500000 |
| 37 | self.lr= 1e-3 |
| 38 | self.save_checkpoint=True |
| 39 | self.save_checkpoint_every_x_epoch=10 |
| 40 | self.with_tensorboard=True |
| 41 | self.with_visualizer=False |
| 42 | self.viewer_config_path=os.path.join(SCRIPT_DIR, "./configs/strand_vae_train.toml") |
| 43 | |
| 44 | |
| 45 |