HelixMind-ckpts / config.json
dheraingoud's picture
Upload config.json with huggingface_hub
71f09be verified
{
"num_features": 42,
"batch_size": 128,
"epochs": 60,
"lr": 0.001,
"weight_decay": 0.01,
"warmup_frac": 0.05,
"grad_clip": 1.0,
"patience": 10,
"loss_alpha": 0.55,
"loss_beta": 0.1,
"loss_gamma": 0.15,
"loss_delta": 0.1,
"dropout": 0.15,
"cnn_channels": 128,
"feat_hidden": 256,
"feat_out": 128,
"fusion_dim": 384,
"head_dim": 192,
"version": "3.0",
"best_spearman": 0.6571680304412763,
"total_epochs_trained": 53,
"class_weight": 239.96774193548387,
"architecture": "MultiScaleCNN_5branch_128ch + FeatureNet + DualHeads",
"total_params": 3062018
}