ClementP's picture
Push model using huggingface_hub.
3628aa4 verified
raw
history blame
318 Bytes
{
"architecture": "resnet50",
"as_regression": true,
"ema": {
"decay": 0.999
},
"img_size": [
512,
512
],
"lr": 3e-05,
"num_classes": 5,
"optimizer": {
"betas": [
0.9,
0.999
],
"weight_decay": 0.0005
},
"pretrained": true,
"swa": {
"swa_lrs": 3e-05
}
}