cookiefinder
/

FF_CV_Challenge

Model card Files Files and versions

Metrics Training metrics Community

cookiefinder commited on Mar 18, 2023

Commit

a7694f4

·

1 Parent(s): 5253d8d

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -16,7 +16,7 @@ Used in 'aiornot' classification competition on HuggingFace. Performs binary cla
 Use the code below to get started with the model.
-    ckpt = torch.load('last.ckpt', map_location=device)
     weights = ckpt['model']
     model = torchvision.models.vgg19_bn().to(device)
     model.load_state_dict(weights)

 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)