Commit Graph

4 Commits (6688d0669f31d5656e7d5eef9ca1117f6fd67d6f)

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