Skip to content

Commit

Permalink
update readme
Browse files Browse the repository at this point in the history
  • Loading branch information
innat committed Apr 3, 2024
1 parent d3b279c commit 868cdd9
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -80,6 +80,11 @@ backbone = VideoSwinT(
)
```

Or, we use use the `VideoSwinBackbone` API directly from `from videoswin.backbone`.


**Arbitrary Input Shape**

By default, the video swin officially is trained with input shape of `32, 224, 224, 3`. But, We can load the model with different shape. And also load the pretrained weight partially.

```python
Expand All @@ -88,9 +93,10 @@ model = VideoSwinT(
include_rescaling=False,
num_classes=10,
)
model.load_weights('model.weights.h5', skip_mismatch=True)
model.load_weights('...weights.h5', skip_mismatch=True)
```


**Guides**

- To ensure the keras reimplementation with official torch: [logit comparison](guides/video-swin-transformer-keras-and-torchvision.ipynb)
Expand Down

0 comments on commit 868cdd9

Please sign in to comment.