29 Commits (main)

Author	SHA1	Message	Date
Ross Wightman	0f2803de7a	Move ImageNet metadata (aka info) files to timm/data/_info. Add helper classes to make info available for labelling. Update inference.py for first use.	1 year ago
Fredo Guan	81ca323751	Davit update formatting and fix grad checkpointing (#7) ... fixed head to gap->norm->fc as per convnext, along with option for norm->gap->fc failed tests due to clip convnext models, davit tests passed	1 year ago
Ross Wightman	927f031293	Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models	1 year ago
Ross Wightman	dbe7531aa3	Update scripts to support torch.compile(). Make --results_file arg more consistent across benchmark/validate/inference. Fix #1570	1 year ago
Ross Wightman	05637a4bb0	More inference script changes, arg naming, multiple output fmts at once	1 year ago
Ross Wightman	eceeb9409a	Significant ugprade to inference.py, support for different formats, formatting, etc.	1 year ago
comar	d7e1e7144a	fix: the exeption when topk is 1	3 years ago
comar	ffa90e04d3	fix: the exception not using default topk argument	3 years ago
Csaba Kertesz	5114c214fc	Change the Python interpreter to Python 3.x in the scripts	3 years ago
Ross Wightman	59ec7e6a53	Merge branch 'master' into imagenet21k_datasets_more	3 years ago
Csaba Kertesz	e42b140ade	Add --input-size option to scripts to specify full input dimensions from command-line	3 years ago
Ross Wightman	de6046e213	Initial commit for dataset / parser reorg to support additional datasets / types	3 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 ... with this update one can tune the kind of logs generated by timm but training and inference traces are unchanged	4 years ago
Ross Wightman	0ebcd2eca1	No default img size for inference script. Fixes #109	4 years ago
Ross Wightman	40fea63ebe	Add checkpoint averaging script. Add headers, shebangs, exec perms to all scripts	4 years ago
Ross Wightman	c6b32cbe73	A number of tweaks to arguments, epoch handling, config ... * reorganize train args * allow resolve_data_config to be used with dict args, not just arparse * stop incrementing epoch before save, more consistent naming vs csv, etc * update resume and start epoch handling to match above * stop auto-incrementing epoch in scheduler	5 years ago
Ross Wightman	6fc886acaf	Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py	5 years ago
Ross Wightman	aa4354f466	Big re-org, working towards making pip/module as 'timm'	5 years ago
Ross Wightman	c1a84ecb22	dataset not passed through PrefetchLoader for inference script. Fix #10 ... * also, make top5 configurable for lower class count cases	5 years ago
Ross Wightman	0562b91c38	Add per model crop pct, interpolation defaults, tie it all together ... * create one resolve fn to pull together model defaults + cmd line args * update attribution comments in some models * test update train/validation/inference scripts	5 years ago
Ross Wightman	9c3859fb9c	Uniform pretrained model handling. ... * All models have 'default_cfgs' dict * load/resume/pretrained helpers factored out * pretrained load operates on state_dict based on default_cfg * test all models in validate * schedule, optim factor factored out * test time pool wrapper applied based on default_cfg	5 years ago
Ross Wightman	0bc50e84f8	Lots of refactoring and cleanup. ... * Move 'test time pool' to Module that can be used by any model, remove from DPN * Remove ResNext model file and combine with ResNet * Remove fbresnet200 as it was an old conversion and pretrained performance not worth param count * Cleanup adaptive avgmax pooling and add back conctat variant * Factor out checkpoint load fn	5 years ago
Ross Wightman	1e23727f2f	Update inference script for new loader style	5 years ago
Ross Wightman	31055466fc	Fixup validate/inference script args, fix senet init for better test accuracy	5 years ago
Ross Wightman	5855b07ae0	Initial commit, puting some ol pieces together	5 years ago