File tree Expand file tree Collapse file tree 2 files changed +13
-1
lines changed
Expand file tree Collapse file tree 2 files changed +13
-1
lines changed Original file line number Diff line number Diff line change 22
33## What's New
44
5+ ### Jan 25, 2021
6+ * Add ResNetV2 Big Transfer (BiT) models w/ ImageNet-1k and 21k weights from https://github.com/google-research/big_transfer
7+ * Add official R50+ViT-B/16 hybrid models + weights from https://github.com/google-research/vision_transformer
8+ * Add model defs and weights for DeiT Vision Transformer models from https://github.com/facebookresearch/deit
9+ * Refactor dataset classes into ImageDataset/IterableImageDataset + dataset specific parser classes
10+ * Add Tensorflow-Datasets (TFDS) wrapper to allow use of TFDS image classification sets with train script
11+ * Ex: ` train.py /data/tfds --dataset tfds/oxford_iiit_pet --val-split test --model resnet50 -b 256 --amp --num-classes 37 --opt adamw --lr 3e-4 --weight-decay .001 --pretrained -j 2 `
12+ * Add improved .tar dataset parser that reads images from .tar, folder of .tar files, or .tar within .tar
13+ * Run validation on full ImageNet-21k directly from tar w/ BiT model: ` validate.py /data/fall11_whole.tar --model resnetv2_50x1_bitm_in21k --amp `
14+ * Models in this update should be stable w/ possible exception of ViT/BiT, possibility of some regressions with train/val scripts and dataset handling
15+
516### Jan 3, 2021
617* Add SE-ResNet-152D weights
718 * 256x256 val, 0.94 crop top-1 - 83.75
@@ -132,6 +143,7 @@ A full version of the list below with source links can be found in the [document
132143
133144* Big Transfer ResNetV2 (BiT) - https://arxiv.org/abs/1912.11370
134145* CspNet (Cross-Stage Partial Networks) - https://arxiv.org/abs/1911.11929
146+ * DeiT (Vision Transformer) - https://arxiv.org/abs/2012.12877
135147* DenseNet - https://arxiv.org/abs/1608.06993
136148* DLA - https://arxiv.org/abs/1707.06484
137149* DPN (Dual-Path Network) - https://arxiv.org/abs/1707.01629
Original file line number Diff line number Diff line change 1- __version__ = '0.3.3 '
1+ __version__ = '0.4.0 '
You can’t perform that action at this time.
0 commit comments