From ecce7c86c7f7adbfe99f47c3f68d07d8f4023fb8 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 19 Dec 2021 18:04:52 +0100 Subject: [PATCH] update comments --- classifier.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/classifier.py b/classifier.py index 857f59fbf..2cbf47469 100644 --- a/classifier.py +++ b/classifier.py @@ -8,9 +8,9 @@ Usage-train: Usage-inference: from classifier import * - model = torch.load('best.pt', map_location=torch.device('cpu'))['model'].float() - files = Path('../datasets/mnist/test/7').glob('*.png') - for f in list(files)[:10]: + model = torch.load('path/to/best.pt', map_location=torch.device('cpu'))['model'].float() + files = Path('../datasets/mnist/test/7').glob('*.png') # images from dir + for f in list(files)[:10]: # first 10 images classify(model, size=128, file=f) """