Skip to content

Instantly share code, notes, and snippets.

@mrdbourke

mrdbourke/mac-m3-pytorch-ml-test-3-imdb.md

Created January 9, 2024 00:57
Show Gist options
  • Save mrdbourke/07840add769b365be1836c1e91406a69 to your computer and use it in GitHub Desktop.
Save mrdbourke/07840add769b365be1836c1e91406a69 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
mac-m3-pytorch-ml-test-3-imdb.md
Model Dataset Sequence Size Epochs Num Samples Num Classes Problem Type
DistilBERT (fine-tune top 2 layers + top Transformer block) IMDB 512 5 25,000 train, 25,000 test 2 Text Classification
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment