File size: 602 Bytes
45b5843 3e8bd23 45b5843 |
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 |
{
"architectures": [
"GAN"
],
"auto_map": {
"AutoConfig": "configuration_ganbert.GanBertConfig",
"AutoModel": "ganbert.GAN"
},
"batch_size": 64,
"device": "cuda",
"epochs": 10,
"epsilon": 1e-08,
"learning_rate_discriminator": 1e-05,
"learning_rate_generator": 1e-05,
"model_number": -2,
"model_type": "ganbert",
"noise_size": 100,
"num_hidden_layers_d": 1,
"num_hidden_layers_g": 2,
"num_train_examples": 77450,
"out_dropout_rate": 0.4,
"pos_class_weight": 10,
"torch_dtype": "float32",
"transformers_version": "4.20.1",
"warmup_proportion": 0.1
}
|