Skip to content

Commit

Permalink
Update readme.md
Browse files Browse the repository at this point in the history
  • Loading branch information
@iamhankai
iamhankai authored Aug 27, 2021
1 parent 8fc3639 commit 0c34644
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions tnt_pytorch/readme.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,7 @@ python train.py /path/to/imagenet/ --model tnt_s_patch16_224 -b 256 --pretrain_p
```

## Third-party implementations
1. Pytorch with **ImageNet pretrained models**: https://www.github.com/rwightman/pytorch-image-models/tree/master/timm/models/tnt.py
2. JAX/FLAX: https://github.com/NZ99/transformer_in_transformer_flax
3. MindSpore Code: https://gitee.com/mindspore/mindspore/tree/master/model_zoo/research/cv/TNT and pretrained weights on Oxford-IIIT Pets dataset: https://www.mindspore.cn/resources/hub/details?noah-cvlab/gpu/1.1/tnt_v1.0_oxford_pets
1. Pytorch (timm) with ImageNet pretrained models: https://www.github.com/rwightman/pytorch-image-models/tree/master/timm/models/tnt.py
2. Pytorch (mmclassification) with ImageNet pretrained models: https://github.com/open-mmlab/mmclassification/blob/master/docs/model_zoo.md
3. JAX/FLAX: https://github.com/NZ99/transformer_in_transformer_flax
4. MindSpore Code: https://gitee.com/mindspore/mindspore/tree/master/model_zoo/research/cv/TNT and pretrained weights on Oxford-IIIT Pets dataset: https://www.mindspore.cn/resources/hub/details?noah-cvlab/gpu/1.1/tnt_v1.0_oxford_pets

0 comments on commit 0c34644

Please sign in to comment.