Aman Arora
|
3f028ebc0f
|
import wandb in summary.py
|
3 years ago |
Aman Arora
|
a9e5d9e5ad
|
log loss as before
|
3 years ago |
Aman Arora
|
624c9b6949
|
log to wandb only if using using wandb
|
3 years ago |
Aman Arora
|
00c8e0b8bd
|
Make use of wandb configurable
|
3 years ago |
Aman Arora
|
8e6fb861e4
|
Add wandb support
|
3 years ago |
Ross Wightman
|
779107b693
|
Merge pull request #542 from juntang-zhuang/adabelief
Add Adabelief Optimizer
|
3 years ago |
Juntang Zhuang
|
74366f733c
|
Delete distributed_train_adabelief.sh
|
3 years ago |
Juntang Zhuang
|
1d848f409a
|
Delete args.yaml
|
3 years ago |
juntang
|
addfc7c1ac
|
adabelief
|
3 years ago |
Ross Wightman
|
fb896c0b26
|
Update some comments re preliminary EfficientNet-V2 assumptions
|
3 years ago |
Ross Wightman
|
2b49ab7a36
|
Fix ResNetV2 pretrained classifier issue. Fixes #540
|
3 years ago |
Ross Wightman
|
de9dff933a
|
EfficientNet-V2S preliminary model def (for experimentation)
|
3 years ago |
Ross Wightman
|
d5ed58d623
|
Merge pull request #533 from rwightman/pit_and_vit_update
Addition of PiT models and update/cleanup of ViT, new NFNet weight, TFDS wrapper fix, few misc fixes/updates
|
3 years ago |
Ross Wightman
|
37c71a5609
|
Some further create_optimizer_v2 tweaks, remove some redudnant code, add back safe model str. Benchmark step times per batch.
|
3 years ago |
Ross Wightman
|
2bb65bd875
|
Wrong default_cfg pool_size for L1
|
3 years ago |
Ross Wightman
|
bf2ca6bdf4
|
Merge jax and original weight init
|
3 years ago |
Ross Wightman
|
acbd698c83
|
Update README.md with updates. Small tweak to head_dist handling.
|
3 years ago |
Ross Wightman
|
9071568f0e
|
Add weights for SE NFNet-L0 model, rename nfnet_l0b -> nfnet_l0. 82.75 top-1 @ 288. Add nfnet_l1 model def for training.
|
3 years ago |
Ross Wightman
|
c468c47a9c
|
Add regnety_160 weights from DeiT teacher model, update that and my regnety_032 weights to use higher test size.
|
3 years ago |
Ross Wightman
|
288682796f
|
Update benchmark script to add precision arg. Fix some downstream (DeiT) compat issues with latest changes. Bump version to 0.4.7
|
3 years ago |
Ross Wightman
|
ea9c9550b2
|
Fully move ViT hybrids to their own file, including embedding module. Remove some extra DeiT models that were for benchmarking only.
|
3 years ago |
Ross Wightman
|
a5310a3451
|
Merge remote-tracking branch 'origin/benchmark-fixes-vit_hybrids' into pit_and_vit_update
|
3 years ago |
Ross Wightman
|
7953e5d11a
|
Fix pos_embed scaling for ViT and num_classes != 1000 for pretrained distilled deit and pit models. Fix #426 and fix #433
|
3 years ago |
Ross Wightman
|
a760a4c3f4
|
Some ViT cleanup, merge distilled model with main, fixup torchscript support for distilled models
|
3 years ago |
Ross Wightman
|
0dfc5a66bb
|
Add PiT model from https://github.com/naver-ai/pit
|
3 years ago |
Ross Wightman
|
1ad1645a50
|
Merge branch 'contrastive-master'
|
3 years ago |
Ross Wightman
|
51febd869b
|
Small tweak to tests for tnt model, reorder model imports.
|
3 years ago |
Ross Wightman
|
b27a4e0d88
|
Merge branch 'master' of https://github.com/contrastive/pytorch-image-models into contrastive-master
|
3 years ago |
Ross Wightman
|
2319cbbff2
|
Merge pull request #525 from amaarora/spp
Add `ActivationStatsHook` to allow extracting activation stats for Signal Propogation Plots
|
3 years ago |
Aman Arora
|
6b18061773
|
Add GIST to docstring for quick access
|
3 years ago |
contrastive
|
809271b0f3
|
Update test_models.py
|
3 years ago |
contrastive
|
de86314655
|
Update TNT
|
3 years ago |
Aman Arora
|
92b1db9a79
|
update docstrings and add check on and
|
3 years ago |
Aman Arora
|
b85be24054
|
update to work with fnmatch
|
3 years ago |
contrastive
|
cfc15283a4
|
Update TNT url
|
3 years ago |
contrastive
|
4a09bc851e
|
Add TNT model
|
3 years ago |
Aman Arora
|
20626e8387
|
Add to extract stats for SPP
|
3 years ago |
Ross Wightman
|
a2727c1bf7
|
Merge pull request #510 from rwightman/dependabot/pip/jinja2-2.11.3
Bump jinja2 from 2.11.2 to 2.11.3
|
4 years ago |
Ross Wightman
|
e2e3290fbf
|
Add '--experiment' to train args for fixed exp name if desired, 'train' not added to output folder if specified.
|
4 years ago |
Ross Wightman
|
cf5fec5047
|
Cleanup experimental vit weight init a bit
|
4 years ago |
dependabot[bot]
|
b36099ace3
|
Bump jinja2 from 2.11.2 to 2.11.3
Bumps [jinja2](https://github.com/pallets/jinja) from 2.11.2 to 2.11.3.
- [Release notes](https://github.com/pallets/jinja/releases)
- [Changelog](https://github.com/pallets/jinja/blob/master/CHANGES.rst)
- [Commits](https://github.com/pallets/jinja/compare/2.11.2...2.11.3)
Signed-off-by: dependabot[bot] <support@github.com>
|
4 years ago |
Ross Wightman
|
f42f1df26c
|
Improve evenness of per-worker split for validation set with TFDS
|
4 years ago |
Ross Wightman
|
cbcb76d72c
|
Should have included Conv2d layers in original weight init. Lets see what the impact is...
|
4 years ago |
Ross Wightman
|
4de57ccf01
|
Add weight init scheme that's closer to JAX impl
|
4 years ago |
Ross Wightman
|
14ac4abf74
|
Change huggingface hub revision delimiter to '@', add hf_hub reference for eca_nfnet_l0 model as an example.
|
4 years ago |
Ross Wightman
|
740f32c96a
|
Add ECA-NFNet-L0 weights and update model name. Update README and bump version to 0.4.6
|
4 years ago |
Ross Wightman
|
5e2e4e7fb6
|
Merge pull request #502 from khatami-mehrdad/efficientnet_dg
adding efficientnet_el, efficientnet_es_pruned and efficientnet_el_pruned pre-trained models
|
4 years ago |
Ross Wightman
|
3eac7dc5a3
|
Merge pull request #501 from rwightman/hf_hub_revisit
Support for huggingface hub via create_model and default_cfgs.
|
4 years ago |
Ross Wightman
|
45c048ba13
|
A few minor fixes and bit more cleanup on the huggingface hub integration.
|
4 years ago |
Mehrdad
|
97f49d163c
|
adding efficientnet_el, efficientnet_es_prunedefficientnet_es_pruned and efficientnet_el_pruned pretrained models to timm/models
|
4 years ago |