code-explain-viz / configs /model_config.json
hmnshudhmn24's picture
Upload 12 files
b91943a verified
{
"base_model": "Salesforce/codet5-small",
"max_input_length": 512,
"max_target_length": 256,
"train_epochs": 3,
"train_batch_size": 8,
"learning_rate": 3e-05
}