Commit Graph

1399 Commits (fixes-syncbn_pretrain_cfg_resolve)
 

Author SHA1 Message Date
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
2 years ago
Jakub Kaczmarzyk db64393c0d
use `Image.Resampling` namespace for PIL mapping (#1256)
2 years ago
Ross Wightman db8e33c69f
Merge pull request #1294 from xwang233/add-aot-autograd
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
2 years ago
Yonghye Kwon 57f8361a01
fix a function parameter typo(cropt_pct -> crop_pct)
2 years ago
Ross Wightman fd360ac951
Merge pull request #1266 from kaczmarj/enh/no-star-imports
2 years ago
Ross Wightman 004fb5f8c4
Merge pull request #1265 from kaczmarj/enh/argparse-groups
2 years ago
Jakub Kaczmarzyk ce5578bc3a replace star imports with imported names
2 years ago
Jakub Kaczmarzyk dcad288fd6 use argparse groups to group arguments
2 years ago
Jakub Kaczmarzyk e1e4c9bbae rm whitespace
2 years ago
Ross Wightman 20a1fa63f8 Make dev version 0.6.2.dev0 for pypi pre
2 years ago
Ross Wightman 3409bb9156
Update README.md
2 years ago
Ross Wightman 137292644b
Update README.md
2 years ago
Ross Wightman 14f370ca1d
Merge pull request #1259 from rwightman/swin_v2
2 years ago
Ross Wightman 347308faad Update README.md, version to 0.6.2
2 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
2 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
2 years ago
Ross Wightman 27c42f0830 Fix torchscript use for offician Swin-V2, add support for non-square window/shift to WindowAttn/Block
2 years ago
Ross Wightman 2f2b22d8c7 Disable nvfuser fma / opt level overrides per #1244
2 years ago
Ross Wightman c0211b0bf7 Swin-V2 test fixes, typo
2 years ago
Ross Wightman 9a86b900fa Official SwinV2 models
2 years ago
Ross Wightman d07d015173
Merge pull request #1249 from okojoalg/sequencer
2 years ago
Ross Wightman d30685c283
Merge pull request #1251 from hankyul2/fix-multistep-scheduler
2 years ago
han a16171335b fix: change milestones to decay-milestones
2 years ago
Ross Wightman 39b725e1c9 Fix tests for rank-4 output where feature channels dim is -1 (3) and not 1
2 years ago
Ross Wightman 78a32655fa Fix poolformer group_matcher to merge proj downsample with previous block, support coarse
2 years ago
Ross Wightman d79f3d9d1e Fix torchscript use for sequencer, add group_matcher, forward_head support, minor formatting
2 years ago
Ross Wightman 37b6920df3 Fix group_matcher regex for regnet.py
2 years ago
okojoalg 93a79a3dd9 Fix num_features in Sequencer
2 years ago
okojoalg 2fec08e923 Add Sequencer to non std filters
2 years ago
han 57a988df30 fix: multistep lr decay epoch bugs
2 years ago
okojoalg 578d52e752 Add Sequencer
2 years ago
Ross Wightman 6d4665bb52
Merge pull request #1245 from rwightman/vit_relpos_refactor
2 years ago
Ross Wightman f5ca4141f7 Adjust arg order for recent vit model args, add a few comments
2 years ago
Ross Wightman 41dc49a337 Vision Transformer refactoring and Rel Pos impl
2 years ago
Ross Wightman b7cb8d0337 Add Swin-V2 Small-NS weights (83.5 @ 224). Add layer scale like 'init_values' via post-norm LN weight scaling
2 years ago
Ross Wightman 001688dabf
Merge pull request #1233 from jjsjann123/nhwc_cond_conv2d
2 years ago
jjsjann123 f88c606fcf fixing channels_last on cond_conv2d; update nvfuser debug env variable
2 years ago
Ross Wightman 7d235c5a5f
Merge pull request #1230 from donglixp/patch-1
2 years ago