Skip to content
This repository has been archived by the owner on Oct 23, 2023. It is now read-only.

Commit

Permalink
Update model urls
Browse files Browse the repository at this point in the history
Summary: Update model urls from previous GitHub repo `pytorch/live` to `facebookresearch/playtorch`

Reviewed By: justinhaaheim

Differential Revision: D39680692

fbshipit-source-id: e5ed43e9fad4cfcb4275b7bace7cf8ce994859f8
  • Loading branch information
raedle committed Nov 20, 2022
1 parent d03b8a3 commit bfd6a97
Showing 1 changed file with 8 additions and 7 deletions.
15 changes: 8 additions & 7 deletions react-native-template-pytorch-live/template/src/Models.ts
Original file line number Diff line number Diff line change
Expand Up @@ -13,41 +13,42 @@ export const ImageClassificationModels: ModelInfo[] = [
{
name: 'ResNet 18',
model:
'https://github.com/pytorch/live/releases/download/v0.1.0/resnet18.ptl',
'https://github.com/facebookresearch/playtorch/releases/download/v0.1.0/resnet18.ptl',
},
{
name: 'MobileNet V3 Small',
model:
'https://github.com/pytorch/live/releases/download/v0.1.0/mobilenet_v3_small.ptl',
'https://github.com/facebookresearch/playtorch/releases/download/v0.1.0/mobilenet_v3_small.ptl',
},
{
name: 'MobileNet V3 Large',
model:
'https://github.com/pytorch/live/releases/download/v0.1.0/mobilenet_v3_large.ptl',
'https://github.com/facebookresearch/playtorch/releases/download/v0.1.0/mobilenet_v3_large.ptl',
},
];

export const MultiClassClassificationModels: ModelInfo[] = [
{
name: 'MNIST',
model: 'https://github.com/pytorch/live/releases/download/v0.1.0/mnist.ptl',
model:
'https://github.com/facebookresearch/playtorch/releases/download/v0.1.0/mnist.ptl',
},
];

export const ObjectDetectionModels: ModelInfo[] = [
{
name: 'DETR',
model:
'https://github.com/pytorch/live/releases/download/v0.1.0/detr_resnet50.ptl',
'https://github.com/facebookresearch/playtorch/releases/download/v0.1.0/detr_resnet50.ptl',
},
];

export const NLPModels: ModelInfo[] = [
{
name: 'DistilBertQA',
model:
'https://github.com/pytorch/live/releases/download/v0.1.0/bert_qa.ptl',
'https://github.com/facebookresearch/playtorch/releases/download/v0.1.0/bert_qa.ptl',
vocab:
'https://github.com/pytorch/live/releases/download/v0.1.0/BERTVocab.json',
'https://github.com/facebookresearch/playtorch/releases/download/v0.1.0/BERTVocab.json',
},
];

0 comments on commit bfd6a97

Please sign in to comment.