{ | |
"architectures": [ | |
"GAN" | |
], | |
"auto_map": { | |
"AutoConfig": "configuration_ganbert.GanBertConfig", | |
"AutoModelForSequenceClassification": "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 | |
} | |