19 Commits (ec0dd4053a75e5eb458bff5d4e4af94aa2fc7326)

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	0161de0127	Switch RandoErasing back to on GPU normal sampling	5 years ago
Ross Wightman	31453b039e	Update Auto/RandAugment comments, README, more. ... * Add a weighted choice option for RandAugment * Adjust magnitude noise/std naming, config	5 years ago
Ross Wightman	4243f076f1	Adding RandAugment to AutoAugment impl, some tweaks to AA included	5 years ago
Ross Wightman	4002c0d4ce	Fix AutoAugment abs translate calc	5 years ago
Ross Wightman	c06274e5a2	Add note on random selection of magnitude value	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	6946281fde	Experimenting with random erasing changes	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	7a92caa560	Add basic image folder style dataset to read directly out of tar files, example in validate.py	5 years ago
Ross Wightman	6cdf35e670	Add explicit half/fp16 support to loader and validation script	5 years ago
Ross Wightman	65a634626f	Switch random erasing to doing normal_() on CPU to avoid instability, remove a debug print	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	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