Firoj112's picture
Upload VQA model in safetensors format after training
2602d5a verified
{
"model_type": "densenet121",
"dataset": "SimulaMet-HOST/Kvasir-VQA",
"training_args": {
"batch_size": 32,
"num_epochs": 1,
"learning_rate": 0.0001
}
}