| { |
| "model_type": "alphadiffract", |
| "architecture": "ConvNeXt1D-MultiTask", |
| "backbone": { |
| "dim_in": 8192, |
| "channels": [80, 80, 80], |
| "kernel_sizes": [100, 50, 25], |
| "strides": [5, 5, 5], |
| "dropout_rate": 0.3, |
| "ramped_dropout_rate": false, |
| "block_type": "convnext", |
| "pooling_type": "average", |
| "final_pool": true, |
| "use_batchnorm": false, |
| "activation": "gelu", |
| "output_type": "flatten", |
| "layer_scale_init_value": 0.0, |
| "drop_path_rate": 0.3 |
| }, |
| "heads": { |
| "head_dropout": 0.5, |
| "cs_hidden": [2300, 1150], |
| "sg_hidden": [2300, 1150], |
| "lp_hidden": [512, 256] |
| }, |
| "tasks": { |
| "num_cs_classes": 7, |
| "num_sg_classes": 230, |
| "num_lp_outputs": 6, |
| "lp_bounds_min": [0.0, 0.0, 0.0, 0.0, 0.0, 0.0], |
| "lp_bounds_max": [500.0, 500.0, 500.0, 180.0, 180.0, 180.0], |
| "bound_lp_with_sigmoid": true |
| }, |
| "training": { |
| "optimizer": "AdamW", |
| "lr": 0.0002, |
| "weight_decay": 0.01, |
| "scheduler": "CyclicLR", |
| "scheduler_mode": "triangular2", |
| "batch_size": 64, |
| "max_epochs": 100, |
| "precision": "float32", |
| "gemd_mu": 1.0, |
| "lambda_cs": 1.0, |
| "lambda_sg": 1.0, |
| "lambda_lp": 1.0 |
| }, |
| "preprocessing": { |
| "input_length": 8192, |
| "floor_at_zero": true, |
| "normalize_range": [0.0, 100.0], |
| "noise_poisson_range": [1.0, 100.0], |
| "noise_gaussian_range": [0.001, 0.1] |
| } |
| } |
|
|
