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