Unverified Commit 7d956458 authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Update comments with the correct recommended endpoint. (#4789)

parent 598fed26
...@@ -3,7 +3,7 @@ This file is part of the private API. Please do not refer to any variables defin ...@@ -3,7 +3,7 @@ This file is part of the private API. Please do not refer to any variables defin
removed on future versions without warning. removed on future versions without warning.
""" """
# This will eventually be replaced with a call at torchvision.datasets.find("imagenet").info.categories # This will eventually be replaced with a call at torchvision.datasets.info("imagenet").categories
_IMAGENET_CATEGORIES = [ _IMAGENET_CATEGORIES = [
"tench", "tench",
"goldfish", "goldfish",
...@@ -1007,7 +1007,7 @@ _IMAGENET_CATEGORIES = [ ...@@ -1007,7 +1007,7 @@ _IMAGENET_CATEGORIES = [
"toilet tissue", "toilet tissue",
] ]
# To be replaced with torchvision.datasets.find("coco").info.categories # To be replaced with torchvision.datasets.info("coco").categories
_COCO_CATEGORIES = [ _COCO_CATEGORIES = [
"__background__", "__background__",
"person", "person",
...@@ -1102,7 +1102,7 @@ _COCO_CATEGORIES = [ ...@@ -1102,7 +1102,7 @@ _COCO_CATEGORIES = [
"toothbrush", "toothbrush",
] ]
# To be replaced with torchvision.datasets.find("voc").info.categories # To be replaced with torchvision.datasets.info("voc").categories
_VOC_CATEGORIES = [ _VOC_CATEGORIES = [
"__background__", "__background__",
"aeroplane", "aeroplane",
...@@ -1127,7 +1127,7 @@ _VOC_CATEGORIES = [ ...@@ -1127,7 +1127,7 @@ _VOC_CATEGORIES = [
"tvmonitor", "tvmonitor",
] ]
# To be replaced with torchvision.datasets.find("kinetics400").info.categories # To be replaced with torchvision.datasets.info("kinetics400").categories
_KINETICS400_CATEGORIES = [ _KINETICS400_CATEGORIES = [
"abseiling", "abseiling",
"air drumming", "air drumming",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment