Kumo-Tabular / config.json
aw777's picture
Add model configuration for download tracking
3c3e10b verified
Raw History Blame
423 Bytes
{
"model_type": "kumo_tabular",
"default_size": "large",
"checkpoint_files": {
"small": {
"classification": "small/classifier.pt",
"regression": "small/regressor.pt"
},
"medium": {
"classification": "medium/classifier.pt",
"regression": "medium/regressor.pt"
},
"large": {
"classification": "large/classifier.pt",
"regression": "large/regressor.pt"
}
}
}