CoolFace
Modelpublic

nailarais1/image-classifier-efficientnet

sourceHugging Faceupdated 2mo agoView on Hugging Face
1likes37downloads
training_config.json35 linesDownload Raw Back to root
1{2  "learning_rate": 0.001,3  "hidden_units": 512,4  "epochs": 3,5  "batch_size": 32,6  "optimizer": "AdamW",7  "scheduler": {8    "type": "StepLR",9    "step_size": 5,10    "gamma": 0.111  },12  "loss": "CrossEntropyLoss",13  "best_validation_accuracy_percent": 94.38,14  "normalization": {15    "mean": [16      0.485,17      0.456,18      0.40619    ],20    "std": [21      0.229,22      0.224,23      0.22524    ]25  },26  "train_transforms": [27    "RandomResizedCrop(224)",28    "RandomHorizontalFlip",29    "ColorJitter(brightness=0.2,contrast=0.2,saturation=0.2,hue=0.1)"30  ],31  "validation_transforms": [32    "Resize(256)",33    "CenterCrop(224)"34  ]35}