(path_or_dict)
| 21 | |
| 22 | |
| 23 | def load_config(path_or_dict): |
| 24 | defaults_image_v1 = { |
| 25 | 'model': { |
| 26 | 'patch_size': 1, |
| 27 | 'augment_wrapper': True, |
| 28 | 'mapping_cond_dim': 0, |
| 29 | 'unet_cond_dim': 0, |
| 30 | 'cross_cond_dim': 0, |
| 31 | 'cross_attn_depths': None, |
| 32 | 'skip_stages': 0, |
| 33 | 'has_variance': False, |
| 34 | }, |
| 35 | 'optimizer': { |
| 36 | 'type': 'adamw', |
| 37 | 'lr': 1e-4, |
| 38 | 'betas': [0.95, 0.999], |
| 39 | 'eps': 1e-6, |
| 40 | 'weight_decay': 1e-3, |
| 41 | }, |
| 42 | } |
| 43 | defaults_image_transformer_v1 = { |
| 44 | 'model': { |
| 45 | 'd_ff': 0, |
| 46 | 'augment_wrapper': False, |
| 47 | 'skip_stages': 0, |
| 48 | 'has_variance': False, |
| 49 | }, |
| 50 | 'optimizer': { |
| 51 | 'type': 'adamw', |
| 52 | 'lr': 5e-4, |
| 53 | 'betas': [0.9, 0.99], |
| 54 | 'eps': 1e-8, |
| 55 | 'weight_decay': 1e-4, |
| 56 | }, |
| 57 | } |
| 58 | defaults_image_transformer_v2 = { |
| 59 | 'model': { |
| 60 | 'mapping_width': 512, |
| 61 | 'mapping_depth': 2, |
| 62 | 'mapping_d_ff': None, |
| 63 | 'mapping_cond_dim': 0, |
| 64 | 'mapping_dropout_rate': 0., |
| 65 | 'd_ffs': None, |
| 66 | 'self_attns': None, |
| 67 | 'dropout_rate': None, |
| 68 | 'augment_wrapper': False, |
| 69 | 'skip_stages': 0, |
| 70 | 'has_variance': False, |
| 71 | }, |
| 72 | 'optimizer': { |
| 73 | 'type': 'adamw', |
| 74 | 'lr': 5e-4, |
| 75 | 'betas': [0.9, 0.99], |
| 76 | 'eps': 1e-8, |
| 77 | 'weight_decay': 1e-4, |
| 78 | }, |
| 79 | } |
| 80 | defaults_image_transformer_v2_cond_keyu = { |
nothing calls this directly
no test coverage detected