Skip to content

Commit fb706cd

Browse files
authored
Update timm.md (#1194)
1 parent 1e71477 commit fb706cd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docs/hub/timm.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ model.eval()
5858
transform = create_transform(**resolve_data_config(model.pretrained_cfg, model=model))
5959

6060
# Get the labels from the model config
61-
labels = model.pretrained_cfg['labels']
61+
labels = model.pretrained_cfg['label_names']
6262
top_k = min(len(labels), 5)
6363

6464
# Use your own image file here...

0 commit comments

Comments
 (0)