diff --git a/README.md b/README.md index 8ec0ea0..284421e 100644 --- a/README.md +++ b/README.md @@ -22,7 +22,7 @@ from paddle.vision import Cifar100 from ppim import rexnet_100 -# Load Model +# Load the model model, val_transforms = rexnet_100(pretrained=True) # Use the PaddleHapi Model