Commit Graph

10 Commits (7419e9835f49de6f35c5343e7033596d2a61bfe8)

Author SHA1 Message Date
Ross Wightman db8ad25a23 MobileNetV3 appears correct based on paper update, cleaned up comments and compacted last block def
5 years ago
Ross Wightman 20d66beead Move RMSpropTF another step closer to Tensorflow impl
5 years ago
Ross Wightman 89147a91e6 Remove adabound optimizer, never got it working well on larger datasets
5 years ago
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