File size: 321 Bytes
b5084fd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
{
"model_type": "swin-transformer",
"architecture": "CustomSwinTransformer",
"num_classes": 7,
"emotions": [
"Angry",
"Disgust",
"Fear",
"Happy",
"Neutral",
"Sad",
"Surprise"
],
"input_size": [
224,
224
],
"framework": "pytorch",
"license": "mit"
} |