10 Commits (7622015258a16e2d9291576e98530495b0d2ade6)

Author	SHA1	Message	Date
Ross Wightman	4666cc9aed	Add --pin-mem arg to enable dataloader pin_memory (showing more benefit in some scenarios now), also add --torchscript arg to validate.py for testing models with jit.script	5 years ago
Ross Wightman	b750b76f67	More AutoAugment work. Ready to roll...	5 years ago
Ross Wightman	25d2088d9e	Working on auto-augment	5 years ago
Ross Wightman	f37e633e9b	Merge remote-tracking branch 'origin/re-exp' into opt	5 years ago
Ross Wightman	daeaa113e2	Add initial sotabench attempt. Split create_transform out of create_loader. Update requirements.txt	5 years ago
Ross Wightman	66634d2200	Add support to split random erasing blocks into randomly selected number with --recount arg. Fix random selection of aspect ratios.	5 years ago
Ross Wightman	dfa9298b4e	Add MixNet (https://arxiv.org/abs/1907.09595) with pretrained weights converted from Tensorflow impl ... * refactor 'same' convolution and add helper to use MixedConv2d when needed * improve performance of 'same' padding for cases that can be handled statically * add support for extra exp, pw, and dw kernel specs with grouping support to decoder/string defs for MixNet * shuffle some args for a bit more consistency, a little less clutter overall in gen_efficientnet.py	5 years ago
Ross Wightman	6cdf35e670	Add explicit half/fp16 support to loader and validation script	5 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	aa4354f466	Big re-org, working towards making pip/module as 'timm'	5 years ago