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: D39680688

fbshipit-source-id: b929cae1939fe571c1e45f0dfd564e4036f467ad
  • Loading branch information
raedle committed Nov 20, 2022
1 parent bfd6a97 commit edf43de
Showing 1 changed file with 10 additions and 9 deletions.
19 changes: 10 additions & 9 deletions react-native-pytorch-core/example/src/Models.ts
Original file line number Diff line number Diff line change
Expand Up @@ -13,57 +13,58 @@ 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 ImageGenerationModels: ModelInfo[] = [
{
name: 'AnimeGANv2',
model:
'https://github.com/pytorch/live/releases/download/v0.2.0-rc.0/animegan2_face_paint_512_v2.ptl',
'https://github.com/facebookresearch/playtorch/releases/download/v0.2.0-rc.0/animegan2_face_paint_512_v2.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',
},
];

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

0 comments on commit edf43de

Please sign in to comment.