model_config.json

{
  "model_name": "bert-base-uncased",
  "language": "english",
  "hs_code_digits": 6,
  "hidden_size": 768,
  "max_seq_len": 128,
  "training_mode": "full_ft_supcon_arcface_mixup_swa",
  "dropout": 0.15,
  "arcface_scale": 30.0,
  "arcface_margin": 0.3,
  "supcon_epochs": 5,
  "classes": {
    "n2": 97,
    "n4": 1266,
    "n6": 6750
  },
  "best_val_acc_6": 90.11904761904762,
  "architecture": "CascadedClassifier_ArcFace_3level"
}