Commit Graph

107 Commits (e9aac412de82310e6905992e802b1ee4dc52b5d1)

Author SHA1 Message Date
Ross Wightman dfa9298b4e Add MixNet (https://arxiv.org/abs/1907.09595) with pretrained weights converted from Tensorflow impl
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
6 years ago
Ross Wightman 65a634626f Switch random erasing to doing normal_() on CPU to avoid instability, remove a debug print
6 years ago
Ross Wightman c6b32cbe73 A number of tweaks to arguments, epoch handling, config
6 years ago
Ross Wightman 6fc886acaf Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py
6 years ago
Ross Wightman aa4354f466 Big re-org, working towards making pip/module as 'timm'
6 years ago