Commit Graph

7 Commits (8fbd62a169c912a02cf80820a9b2c30168a8b8b0)

Author SHA1 Message Date
Ross Wightman 8fbd62a169 Exclude batchnorm and bias params from weight_decay by default
5 years ago
Ross Wightman bc264269c9 Morph mnasnet impl into a generic mobilenet that covers Mnasnet, MobileNetV1/V2, ChamNet, FBNet, and related
5 years ago
Ross Wightman db1fe34d0c Update a few comment, add some references
5 years ago
Ross Wightman 9c3859fb9c Uniform pretrained model handling.
5 years ago
Ross Wightman 2295cf56c2 Add some Nvidia performance enhancements (prefetch loader, fast collate), and refactor some of training and model fact/transforms
5 years ago
Ross Wightman 9d927a389a Add adabound, random erasing
5 years ago
Ross Wightman 5855b07ae0 Initial commit, puting some ol pieces together
6 years ago