@@ -48,4 +48,6 @@ class AnnotationType(BaseChoiceType):
INAT20_U_OUT_CLASS = partial(INAT20_Annotations)
INAT20_NOISY_OUT_CLASS = partial(INAT20_Annotations)
+ IMAGENET_TOP_INAT20 = partial(INET_Annotations)
+
Default = CUB200
