File size: 1,077 Bytes
da99c05 34a4bae da99c05 369143b da99c05 2e0bea9 d6b9350 34a4bae da99c05 6d1c952 da99c05 d6b9350 da99c05 6d1c952 da99c05 f9b1265 da99c05 5aeca1a d6b9350 da99c05 6d1c952 da99c05 3e80af5 da99c05 f9b1265 da99c05 34a4bae e30a094 da99c05 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
{
"_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"
}
|