config.json

{
  "_name_or_path": "./exp/fox_nbc/checkpoints\\best",
  "architectures": [
    "CustomModel"
  ],
  "base_exp_dir": "./exp/fox_nbc/",
  "model": {
    "cls": {
      "combined_dim": 128,
      "combined_input": 1024,
      "n_layers": 2,
      "num_classes": 2,
      "skip_in": [
        80
      ],
      "weight_norm": true
    },
    "freq": {
      "n_layers": 2,
      "skip_in": [
        80
      ],
      "tfidf_hidden_dim": 512,
      "tfidf_input_dim": 8145,
      "tfidf_output_dim": 128,
      "weight_norm": true
    },
    "pos": {
      "hidden_dim": 256,
      "input_dim": 300,
      "n_layers": 2,
      "output_dim": 128,
      "skip_in": [
        80
      ],
      "weight_norm": true
    }
  },
  "model_type": "headlineclassifier",
  "torch_dtype": "float32",
  "train": {
    "anneal_end": 5,
    "batch_size": 32,
    "end_iter": 10,
    "learning_rate": 2e-05,
    "learning_rate_alpha": 0.05,
    "save_freq": 1,
    "val_freq": 1,
    "warm_up_end": 2
  },
  "transformers_version": "4.47.0"
}