Commit Graph

26 Commits (8016ac3569d09a0098a7386b96244dde055227b0)

Author SHA1 Message Date
Ross Wightman dbe7531aa3 Update scripts to support torch.compile(). Make --results_file arg more consistent across benchmark/validate/inference. Fix #1570
2 years ago
Ross Wightman 05637a4bb0 More inference script changes, arg naming, multiple output fmts at once
2 years ago
Ross Wightman eceeb9409a Significant ugprade to inference.py, support for different formats, formatting, etc.
2 years ago
comar d7e1e7144a fix: the exeption when topk is 1
4 years ago
comar ffa90e04d3 fix: the exception not using default topk argument
4 years ago
Csaba Kertesz 5114c214fc Change the Python interpreter to Python 3.x in the scripts
4 years ago
Ross Wightman 59ec7e6a53 Merge branch 'master' into imagenet21k_datasets_more
4 years ago
Csaba Kertesz e42b140ade Add --input-size option to scripts to specify full input dimensions from command-line
4 years ago
Ross Wightman de6046e213 Initial commit for dataset / parser reorg to support additional datasets / types
4 years ago
Matthijs Hollemans f04bdc8c8e don't forget this file
4 years ago
Ross Wightman e8ca45854c More models in sotabench, more control over sotabench run, dataset filename extraction consistency
4 years ago
Ross Wightman 7995295968 Merge branch 'logger' into features. Change 'logger' to '_logger'.
4 years ago
Ross Wightman 6c17d57a2c Fix some attributions, add copyrights to some file docstrings
4 years ago
Antoine Broyelle 78fa0772cc Leverage python hierachical logger
5 years ago
Ross Wightman 0ebcd2eca1 No default img size for inference script. Fixes #109
5 years ago
Ross Wightman 40fea63ebe Add checkpoint averaging script. Add headers, shebangs, exec perms to all scripts
5 years ago
Ross Wightman c6b32cbe73 A number of tweaks to arguments, epoch handling, config
6 years ago
Ross Wightman 6fc886acaf Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py
6 years ago
Ross Wightman aa4354f466 Big re-org, working towards making pip/module as 'timm'
6 years ago
Ross Wightman c1a84ecb22 dataset not passed through PrefetchLoader for inference script. Fix #10
6 years ago
Ross Wightman 0562b91c38 Add per model crop pct, interpolation defaults, tie it all together
6 years ago
Ross Wightman 9c3859fb9c Uniform pretrained model handling.
6 years ago
Ross Wightman 0bc50e84f8 Lots of refactoring and cleanup.
6 years ago
Ross Wightman 1e23727f2f Update inference script for new loader style
6 years ago
Ross Wightman 31055466fc Fixup validate/inference script args, fix senet init for better test accuracy
6 years ago
Ross Wightman 5855b07ae0 Initial commit, puting some ol pieces together
6 years ago