From dc6476229069b0b13fce734b5f195fed68ecfb09 Mon Sep 17 00:00:00 2001 From: Ali Hatamizadeh Date: Wed, 5 Jul 2023 07:48:13 -0700 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 6642e8b..b22a10a 100644 --- a/README.md +++ b/README.md @@ -49,7 +49,7 @@ A pretrained FasterViT model with default hyper-parameters can be created as in model_path="/tmp/faster_vit_0.pth.tar") ``` -We can adjust where to download the above FasterViT-0 model by passing `model_path`. +`model_path` is used to set the directory to download the model. We can also simply test the model by passing a dummy input image. The output is the logits: