Model Card for VGG19
VGG19BN Model from torchvision.models
Used in 'aiornot' classification competition on HuggingFace. Performs binary classification to detect whether the image is AI-generated or not.
Model Details
How to Get Started with the Model
Use the code below to get started with the model.
ckpt = torch.load('weights.ckpt', map_location=device)
weights = ckpt['model']
model = torchvision.models.vgg19_bn().to(device)
model.load_state_dict(weights)