Skip to content

Commit af1a68d

Browse files
authored
Update README.md
1 parent 8ceceef commit af1a68d

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

README.md

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,8 @@ The work of many others is present here. I've tried to make sure all source mate
1111
* [Torchvision](https://github.com/pytorch/vision/tree/master/torchvision/models)
1212
* [Cadene's Pretrained Models](https://github.com/Cadene/pretrained-models.pytorch)
1313
* [Myself](https://github.com/rwightman/pytorch-dpn-pretrained)
14-
* LR scheduler ideas from [AllenNLP](https://github.com/allenai/allennlp/tree/master/allennlp/training/learning_rate_schedulers) and [FAIRseq](https://github.com/pytorch/fairseq/tree/master/fairseq/optim/lr_scheduler)
15-
* Random Erasing from [Zhun Zhong](https://github.com/zhunzhong07/Random-Erasing/blob/master/transforms.py)
14+
* LR scheduler ideas from [AllenNLP](https://github.com/allenai/allennlp/tree/master/allennlp/training/learning_rate_schedulers), [FAIRseq](https://github.com/pytorch/fairseq/tree/master/fairseq/optim/lr_scheduler), and SGDR: Stochastic Gradient Descent with Warm Restarts (https://arxiv.org/abs/1608.03983)
15+
* Random Erasing from [Zhun Zhong](https://github.com/zhunzhong07/Random-Erasing/blob/master/transforms.py) (https://arxiv.org/abs/1708.04896)
1616

1717
## Models
1818

@@ -30,12 +30,13 @@ I've included a few of my favourite models, but this is not an exhaustive collec
3030
* DPN (from [me](https://github.com/rwightman/pytorch-dpn-pretrained), weights hosted by Cadene)
3131
* DPN-68, DPN-68b, DPN-92, DPN-98, DPN-131, DPN-107
3232
* Generic MobileNet (from my standalone [GenMobileNet](https://github.com/rwightman/genmobilenet-pytorch)) - A generic model that implements many of the mobile optimized architecture search derived models that utilize similar DepthwiseSeparable and InvertedResidual blocks
33-
* MNASNet B1, A1 (Squeeze-Excite), and Small
34-
* MobileNet-V1
35-
* MobileNet-V2
36-
* MobileNet-V3 (work in progress, validating config)
37-
* ChamNet (details hard to find, currently an educated guess)
38-
* FBNet-C (TODO A/B variants)
33+
* MNASNet B1, A1 (Squeeze-Excite), and Small (https://arxiv.org/abs/1807.11626)
34+
* MobileNet-V1 (https://arxiv.org/abs/1704.04861)
35+
* MobileNet-V2 (https://arxiv.org/abs/1801.04381)
36+
* MobileNet-V3 (https://arxiv.org/abs/1905.02244) -- work in progress, validating config
37+
* ChamNet (https://arxiv.org/abs/1812.08934) -- specific arch details hard to find, currently an educated guess
38+
* FBNet-C (https://arxiv.org/abs/1812.03443) -- TODO A/B variants
39+
* Single-Path NAS (https://arxiv.org/abs/1904.02877) -- pixel1 variant
3940

4041
The full list of model strings that can be passed to model factory via `--model` arg for train, validation, inference scripts:
4142
```

0 commit comments

Comments
 (0)