Commit Graph

4 Commits (7dab6d1ec7466d5eb23ddad4e275ff277d7b8cb4)

Author SHA1 Message Date
Ross Wightman 89147a91e6 Remove adabound optimizer, never got it working well on larger datasets
6 years ago
Ross Wightman bc264269c9 Morph mnasnet impl into a generic mobilenet that covers Mnasnet, MobileNetV1/V2, ChamNet, FBNet, and related
6 years ago
Ross Wightman 9c3859fb9c Uniform pretrained model handling.
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