Skip to content

Commit

Permalink
update readme: add info"
Browse files Browse the repository at this point in the history
  • Loading branch information
innat committed Oct 6, 2023
1 parent e0b2036 commit a65d558
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,8 +10,9 @@ This is a `Keras` reimplementation of [VideoMAE: Masked Autoencoders are Data-Ef

## Pre Trained Self-Supervised Model

```python
The pre-trained video-mae model consist of encoder and deconder module. This models are trained in self-supervised manner on the benchmark dataset.

```python
from videomae import VideoMAE_ViTS16PT

# pre-trained self-supervised model
Expand All @@ -37,6 +38,8 @@ TensorShape([1, 1176, 1536])

## Fine Tuned Model

The fine tuned model is the encoder part of pre-trained model which is used to model for specific target classes.

```python
from videomae import VideoMAE_ViTS16FT

Expand Down

0 comments on commit a65d558

Please sign in to comment.