Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
ahatamiz authored Jul 5, 2023
1 parent d43ee39 commit dc64762
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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:

Expand Down

0 comments on commit dc64762

Please sign in to comment.