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