| { |
| "architectures": [ |
| "FIMImputation" |
| ], |
| "fim_base": "FIM4Science/fim-imputation-base", |
| "global_normalization_times": { |
| "name": "fim.models.blocks.MinMaxNormalization" |
| }, |
| "global_normalization_values": { |
| "name": "fim.models.blocks.MinMaxNormalization" |
| }, |
| "loss_configs": { |
| "loss_scale_drift": 1.0, |
| "loss_scale_latent_embedding": 0.0, |
| "loss_scale_unsuperv_loss": 0.0 |
| }, |
| "model_type": "fim_imputation", |
| "psi_2": { |
| "dim_model": 512, |
| "dim_time": 1023, |
| "dropout": 0.1, |
| "name": "fim.models.blocks.Transformer", |
| "num_encoder_blocks": 4, |
| "num_heads": 8, |
| "residual_mlp": { |
| "dropout": 0.1, |
| "hidden_act": { |
| "name": "torch.nn.SELU" |
| }, |
| "hidden_layers": [ |
| 512 |
| ], |
| "in_features": 512, |
| "name": "fim.models.blocks.MLP", |
| "out_features": 512, |
| "output_act": { |
| "name": "torch.nn.Identity" |
| } |
| } |
| }, |
| "scale_feature_mapping": { |
| "bias": true, |
| "in_features": 9, |
| "name": "torch.nn.Linear", |
| "out_features": 512 |
| }, |
| "torch_dtype": "float32", |
| "transformers_version": "4.51.3", |
| "use_fim_normalization": true |
| } |
|
|
