{ "model_name": "ArcFace_iResNet50_CASIA_FaceV5", "architecture": "iresnet50", "framework": "PyTorch", "task":"face-recognition", "input_size": [112, 112, 3], "num_classes": 512, "pretrained": true, "description": "ArcFace model based on iResNet50 backbone for face recognition. This model has been fine-tuned on the CASIA_FaceV5 dataset.", "training_dataset": "CASIA_FaceV5", "embedding_size": 512, "margin": 0.5, "scale": 64.0, "optimizer": "Adam", "learning_rate": 1e-4, "lr_decay_type": "cos", "weight_decay": 0, "momentum": 0.9, "loss_function": "ArcFaceLoss", "license": "Apache License 2.0", "notes": "This model is trained for face recognition tasks and outputs a 512-dimensional embedding vector. It has been fine-tuned on the CASIA_FaceV5 dataset for improved performance." }