Default Models List

Get a summary of the default models for all the tasks covered in the course.

We'll cover the following

As a reference, we've provided a summarized list of all the tasks covered in the course along with the default models.

NLP

For Natural Language Processing, the following are the default models for respective tasks.

  • Text classification: distilbert-base-uncased-finetuned-sst-2-english
  • Token classification: dslim/bert-base-NER
  • Text summarization: sshleifer/distilbart-cnn-12-6
  • Question answering: distilbert-base-cased-distilled-squad
  • Text generation: gpt2
  • Text similarity: sentence-transformers/all-mpnet-base-v2
  • Translation: t5-base
  • Fill mask: distilroberta-base

Computer vision

The default models for computer vision tasks are as follows:

  • Image classification: google/vit-base-patch16-224
  • Object detection: facebook/detr-resnet-50
  • Segmentation: facebook/detr-resnet-50-panoptic

Create a free account to view this lesson.

By signing up, you agree to Educative's Terms of Service and Privacy Policy