pytorch-image-models

Commit Graph

Author	SHA1	Message	Date
Ross Wightman	6064d16a2d	Add initial EdgeNeXt import. Significant cleanup / reorg (like ConvNeXt). Fix #1320 * edgenext refactored for torchscript compat, stage base organization * slight refactor of ConvNeXt to match some EdgeNeXt additions * remove use of funky LayerNorm layer in ConvNeXt and just use nn.LayerNorm and LayerNorm2d (permute)	2 years ago
Ross Wightman	7a9c6811c9	Add eps arg to LayerNorm2d, add 'tf' (tensorflow) variant of trunc_normal_ that applies scale/shift after sampling (instead of needing to move a/b)	2 years ago
Ross Wightman	82c311d082	Add more experimental darknet and 'cs2' darknet variants (different cross stage setup, closer to newer YOLO backbones) for train trials.	2 years ago
Ross Wightman	a050fde5cd	Add resnet10t (basic block) and resnet14t (bottleneck) with 1,1,1,1 repeats	2 years ago
Ross Wightman	34f382f8f6	move dataconfig before script, scripting killing metadata now (PyTorch 1.12? just nvfuser?)	2 years ago
Ross Wightman	beef62e7ab	Merge pull request #1317 from rwightman/fixes-syncbn_pretrain_cfg_resolve Fix SyncBatchNorm for BatchNormAc2d, improve resolve_pretrained_cfg behaviour, other mix fixes.	2 years ago
Ross Wightman	e6d7df40ec	no longer a point using kwargs for pretrain_cfg resolve, just pass explicit arg	2 years ago
Ross Wightman	a29fba307d	disable dist_bn when sync_bn active	2 years ago
Ross Wightman	07d0c4ae96	Improve repr for DropPath module	2 years ago
Ross Wightman	e27c16b8a0	Remove unecessary code for synbn guard	2 years ago
Ross Wightman	0da3c9ebbf	Remove SiLU layer in default args that breaks import on old old PyTorch	2 years ago
Ross Wightman	7d657d2ef4	Improve resolve_pretrained_cfg behaviour when no cfg exists, warn instead of crash. Improve usability ex #1311	2 years ago
Ross Wightman	879df47c0a	Support BatchNormAct2d for sync-bn use. Fix #1254	2 years ago
Ross Wightman	7cedc8d474	Follow up to #1256 , fix interpolation warning in auto_autoaugment as well	2 years ago
Ross Wightman	037e5e6c09	Fix #1309 , move wandb init after distributed init, only init on rank == 0 process	2 years ago
Jakub Kaczmarzyk	9e12530433	use utils namespace instead of function/classnames This fixes buggy behavior introduced by https://github.com/rwightman/pytorch-image-models/pull/1266. Related to https://github.com/rwightman/pytorch-image-models/pull/1273.	2 years ago
Jakub Kaczmarzyk	db64393c0d	use `Image.Resampling` namespace for PIL mapping (#1256 ) * use `Image.Resampling` namespace for PIL mapping PIL shows a deprecation warning when accessing resampling constants via the `Image` namespace. The suggested namespace is `Image.Resampling`. This commit updates `_pil_interpolation_to_str` to use the `Image.Resampling` namespace. ``` /tmp/ipykernel_11959/698124036.py:2: DeprecationWarning: NEAREST is deprecated and will be removed in Pillow 10 (2023-07-01). Use Resampling.NEAREST or Dither.NONE instead. Image.NEAREST: 'nearest', /tmp/ipykernel_11959/698124036.py:3: DeprecationWarning: BILINEAR is deprecated and will be removed in Pillow 10 (2023-07-01). Use Resampling.BILINEAR instead. Image.BILINEAR: 'bilinear', /tmp/ipykernel_11959/698124036.py:4: DeprecationWarning: BICUBIC is deprecated and will be removed in Pillow 10 (2023-07-01). Use Resampling.BICUBIC instead. Image.BICUBIC: 'bicubic', /tmp/ipykernel_11959/698124036.py:5: DeprecationWarning: BOX is deprecated and will be removed in Pillow 10 (2023-07-01). Use Resampling.BOX instead. Image.BOX: 'box', /tmp/ipykernel_11959/698124036.py:6: DeprecationWarning: HAMMING is deprecated and will be removed in Pillow 10 (2023-07-01). Use Resampling.HAMMING instead. Image.HAMMING: 'hamming', /tmp/ipykernel_11959/698124036.py:7: DeprecationWarning: LANCZOS is deprecated and will be removed in Pillow 10 (2023-07-01). Use Resampling.LANCZOS instead. Image.LANCZOS: 'lanczos', ``` * use new pillow resampling enum only if it exists	2 years ago
Ross Wightman	db8e33c69f	Merge pull request #1294 from xwang233/add-aot-autograd Add AOT Autograd support	2 years ago
Ross Wightman	2d7ab06503	Move aot-autograd opt after model metadata used to setup data config in benchmark.py	2 years ago
Xiao Wang	ca991c1fa5	add --aot-autograd	2 years ago
Ross Wightman	e4360e6125	Merge pull request #1270 from developer0hye/patch-1 fix a function parameter typo(cropt_pct -> crop_pct)	3 years ago
Yonghye Kwon	57f8361a01	fix a function parameter typo(cropt_pct -> crop_pct)	3 years ago
Ross Wightman	fd360ac951	Merge pull request #1266 from kaczmarj/enh/no-star-imports ENH: replace star imports with imported names in train.py	3 years ago
Ross Wightman	004fb5f8c4	Merge pull request #1265 from kaczmarj/enh/argparse-groups ENH: use argparse groups in training script	3 years ago
Jakub Kaczmarzyk	ce5578bc3a	replace star imports with imported names	3 years ago
Jakub Kaczmarzyk	dcad288fd6	use argparse groups to group arguments	3 years ago
Jakub Kaczmarzyk	e1e4c9bbae	rm whitespace	3 years ago
Ross Wightman	20a1fa63f8	Make dev version 0.6.2.dev0 for pypi pre	3 years ago
Ross Wightman	3409bb9156	Update README.md	3 years ago
Ross Wightman	137292644b	Update README.md	3 years ago
Ross Wightman	14f370ca1d	Merge pull request #1259 from rwightman/swin_v2 Official SwinV2 models	3 years ago
Ross Wightman	347308faad	Update README.md, version to 0.6.2	3 years ago
Ross Wightman	4b30bae67b	Add updated vit_relpos weights, and impl w/ support for official swin-v2 differences for relpos. Add bias control support for MLP layers	3 years ago
Ross Wightman	d4c0588012	Remove persistent buffers from Swin-V2. Change SwinV2Cr cos attn + tau/logit_scale to match official, add ckpt convert, init_value zeros resid LN weight by default	3 years ago
Ross Wightman	27c42f0830	Fix torchscript use for offician Swin-V2, add support for non-square window/shift to WindowAttn/Block	3 years ago
Ross Wightman	2f2b22d8c7	Disable nvfuser fma / opt level overrides per #1244	3 years ago
Ross Wightman	c0211b0bf7	Swin-V2 test fixes, typo	3 years ago
Ross Wightman	9a86b900fa	Official SwinV2 models	3 years ago
Ross Wightman	d07d015173	Merge pull request #1249 from okojoalg/sequencer Add Sequencer	3 years ago
Ross Wightman	d30685c283	Merge pull request #1251 from hankyul2/fix-multistep-scheduler fix: multistep lr decay epoch bugs	3 years ago
han	a16171335b	fix: change milestones to decay-milestones - change argparser option `milestone` to `decay-milestone`	3 years ago
Ross Wightman	39b725e1c9	Fix tests for rank-4 output where feature channels dim is -1 (3) and not 1	3 years ago
Ross Wightman	78a32655fa	Fix poolformer group_matcher to merge proj downsample with previous block, support coarse	3 years ago
Ross Wightman	d79f3d9d1e	Fix torchscript use for sequencer, add group_matcher, forward_head support, minor formatting	3 years ago
Ross Wightman	37b6920df3	Fix group_matcher regex for regnet.py	3 years ago
okojoalg	93a79a3dd9	Fix num_features in Sequencer	3 years ago
okojoalg	2fec08e923	Add Sequencer to non std filters	3 years ago
han	57a988df30	fix: multistep lr decay epoch bugs - add milestones arguments - change decay_epochs to milestones variable	3 years ago
okojoalg	578d52e752	Add Sequencer	3 years ago
Ross Wightman	6d4665bb52	Merge pull request #1245 from rwightman/vit_relpos_refactor Vision Transformer refactoring and Rel Pos impl	3 years ago

1 2 3 4 5 ...

1405 Commits (6064d16a2dfe89b1d3706df338cecfdcee395d1f) All Branches Search

1405 Commits (6064d16a2dfe89b1d3706df338cecfdcee395d1f)

All Branches