Model | Dataset | Sequence Size | Epochs | Num Samples | Num Classes | Problem Type |
---|---|---|---|---|---|---|
SmallTransformer (custom) | IMDB | 200 | 5 | 25,000 train, 25,000 test | 2 | Text Classification |
Created
January 9, 2024 01:02
-
-
Save mrdbourke/30b383d26e11a73ca205e267c5aa1474 to your computer and use it in GitHub Desktop.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment