|
{ |
|
"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 |
|
} |
|
|