yusufcakmak's picture
feat: initial model release - TYArcFace ConvNeXt Product Embeddings
0dfad2f verified
raw
history blame contribute delete
552 Bytes
{
"model_type": "trendyol_arcface",
"architectures": ["TYArcFaceModel"],
"backbone_name": "convnext_base_in22ft1k",
"backbone_features": 1024,
"embedding_dim": 512,
"hidden_size": 49,
"use_arcface_loss": true,
"arcface_scale": 128,
"arcface_margin": 0.25,
"dropout": 0.5,
"input_size": 224,
"normalization": {
"mean": [0.485, 0.456, 0.406],
"std": [0.229, 0.224, 0.225]
},
"task_type": "feature-extraction",
"training_info": {
"epoch": 5,
"global_step": 108,
"pytorch_lightning_version": "1.8.1"
}
}