BGC_abundance_regression_model/regression_config.json

{
  "num_targets": 8,
  "head": "mlp",
  "hidden_size": 512,
  "lstm_layers": 2,
  "cnn_kernel": 3,
  "dropout": 0.1,
  "loss_type": "mse",
  "huber_delta": 1.0,
  "pretrained_dir": "/mnt/zzb/default/Workspace/jyzheng/models/model_3M_2048_v10/final_model",
  "target_stats": {
    "mean": [
      0.27447958386988835,
      0.4005759305638644,
      0.04693631303042901,
      0.8911714305993531,
      0.3163663648439005,
      0.8205084006773268,
      0.010180181093659695,
      1.61296046561147
    ],
    "std": [
      0.8426240485241817,
      1.508517090647771,
      0.25301948095119775,
      1.3785933353680424,
      0.8472574931688738,
      1.0853418828715786,
      0.10228318857711748,
      1.2825537097772675
    ]
  },
  "max_length": 2048
}