update comments

This commit is contained in:
Glenn Jocher 2021-12-19 18:04:52 +01:00
parent d5fc52132a
commit ecce7c86c7

View File

@ -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)
"""