83 Commits (82c311d0821643e8613b0b18f6d0f14088a79459)

Author	SHA1	Message	Date
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
AFLALO, Jonathan Isaac	a7f570c9b7	added MultiEpochsDataLoader	5 years ago
Ross Wightman	64fe37d008	Remove old mean/std helper, rely fully on cmd line or default_cfg now. Fixes #126	5 years ago
Ross Wightman	670c61b28f	Some cutmix/mixup cleanup/fixes	5 years ago
Ross Wightman	b3cb5f3275	Working on CutMix impl as per #8, integrating with Mixup, currently experimenting...	5 years ago
Ross Wightman	d66819d1f3	Indentation mistake. Fixes #81	5 years ago
Ross Wightman	1daa303744	Add support to Dataset for class id mapping file, clean up a bit of old logic. Add results file arg for validation and update script.	5 years ago
Ross Wightman	d9a6a9d0af	Merge pull request #74 from rwightman/augmix-jsd ... AugMix, JSD loss, SplitBatchNorm (Auxiliary BN), and more	5 years ago
Ross Wightman	3eb4a96eda	Update AugMix, JSD, etc comments and references	5 years ago
Ross Wightman	7547119891	Add SplitBatchNorm. AugMix, Rand/AutoAugment, Split (Aux) BatchNorm, Jensen-Shannon Divergence, RandomErasing all working together	5 years ago
Ross Wightman	2e955cfd0c	Update RandomErasing with some improved arg names, tweak to aspect range	5 years ago
Ross Wightman	3cc0f91e23	Fix augmix variable name scope overlap, default non-blended mode	5 years ago
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	3afc2a4dc0	Some cleanup/improvements to AugMix impl: ... * make 'increasing' levels for Contrast, Color, Brightness, Saturation ops * remove recursion from faster blending mix * add config striing parsing for AugMix	5 years ago
Ross Wightman	232ab7fb12	Working on an implementation of AugMix with JensenShannonDivergence loss that's compatible with my AutoAugment and RandAugment impl	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