| { |
| "_name_or_path": "/bigdata/qiuhan/codet5/CodeT5/saved_models/go_generation/codet5_small_all_lr10_bs32_src512_trg256_pat3_e30/checkpoint-best-ppl", |
| "architectures": [ |
| "T5ForConditionalGeneration" |
| ], |
| "bos_token_id": 1, |
| "d_ff": 2048, |
| "d_kv": 64, |
| "d_model": 512, |
| "decoder_start_token_id": 0, |
| "dropout_rate": 0.1, |
| "eos_token_id": 2, |
| "feed_forward_proj": "relu", |
| "gradient_checkpointing": false, |
| "id2label": { |
| "0": "LABEL_0" |
| }, |
| "initializer_factor": 1.0, |
| "is_encoder_decoder": true, |
| "label2id": { |
| "LABEL_0": 0 |
| }, |
| "layer_norm_epsilon": 1e-06, |
| "model_type": "t5", |
| "n_positions": 512, |
| "num_decoder_layers": 6, |
| "num_heads": 8, |
| "num_layers": 6, |
| "output_past": true, |
| "pad_token_id": 0, |
| "relative_attention_num_buckets": 32, |
| |
| "torch_dtype": "float32", |
| "transformers_version": "4.10.2", |
| "use_cache": true, |
| "vocab_size": 32100 |
| } |
|
|
