Commit Graph

7 Commits (4663fc2132589296688503fd4606c0ccf148f81b)

Author SHA1 Message Date
Ross Wightman 8fbd62a169 Exclude batchnorm and bias params from weight_decay by default
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 db1fe34d0c Update a few comment, add some references
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
Ross Wightman 9d927a389a Add adabound, random erasing
6 years ago
Ross Wightman 5855b07ae0 Initial commit, puting some ol pieces together
6 years ago