Skip to content
Switch branches/tags
Go to file
Cannot retrieve contributors at this time
layout: hub_detail
background-class: hub-background
body-class: hub
category: researchers
<!-- Only change fields below(remove this line before submitting a PR). Take inspiration e.g. from -->
title: <REQUIRED: short model name>
summary: <REQUIRED: 1-2 sentences>
image: <REQUIRED: best image to represent your model>
author: <REQUIRED>
tags: <REQUIRED: [tag1, tag2, ...]. Allowed tags are vision, nlp, generative, audio, scriptable>
github-link: <REQUIRED>
github-id: <REQUIRED: top level of repo>
featured_image_1: <OPTIONAL: use no-image if not applicable>
featured_image_2: <OPTIONAL: use no-image if not applicable>
accelerator: <OPTIONAL: Current supported values: "cuda", "cuda-optional">
import torch
torch.hub.load('pytorch/vision', 'resnet18', pretrained=True)

Model Description