7 Commits (af1a68d2e119744501b0a71cf8438399b0933f10)

Author	SHA1	Message	Date
Ross Wightman	780c0a96a4	Change args for RandomErasing so only one required for pixel/color mode	6 years ago
Ross Wightman	76539d905e	Some transform/data/loader refactoring, hopefully didn't break things ... * factor out data related constants to own file * move data related config helpers to own file * add a variant of RandomResizeCrop that randomizes interpolation method * remove old Numpy version of RandomErasing * cleanup torch version of RandomErasing and use it in either GPU loader batch mode or single image cpu Transform	6 years ago
Ross Wightman	c328b155e9	Random erasing crash fix and args pass through	6 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	6 years ago
Ross Wightman	5180f94c7e	Distributed (multi-process) train, multi-gpu single process train, and NVIDIA AMP support	6 years ago
Ross Wightman	5cb1a35c6b	Fixup Resnext, remove alternate shortcut types	6 years ago
Ross Wightman	2295cf56c2	Add some Nvidia performance enhancements (prefetch loader, fast collate), and refactor some of training and model fact/transforms	6 years ago