diff --git a/README.md b/README.md
index 424015f..26440ab 100644
--- a/README.md
+++ b/README.md
@@ -56,7 +56,7 @@ Create the operator via the following factory method
 Refer [Timm Docs](https://fastai.github.io/timmdocs/#List-Models-with-Pretrained-Weights) to get a full list of supported models.
 
 
-​	***num_classes***: *1000*
+​	***num_classes***: *int*
 
 ​	The number of classes. The default value is 1000.
 It is related to model and dataset.