2 Commits (c9a61b7d988ec5907c1a010a0b1bb4ab910f5264)

Author	SHA1	Message	Date
Ross Wightman	20d66beead	Move RMSpropTF another step closer to Tensorflow impl ... * init square_avg with one instead of zero as per TF * match TF order of ops for square_avg accumulation * move LR scaling to momentum buffer accumulator as per TF * add decoupled weight decay flag (not in TF)	2019-05-14 18:35:43 -07:00
Ross Wightman	bc264269c9	Morph mnasnet impl into a generic mobilenet that covers Mnasnet, MobileNetV1/V2, ChamNet, FBNet, and related ... * add an alternate RMSprop opt that applies eps like TF * add bn params for passing through alternates and changing defaults to TF style	2019-04-21 15:54:28 -07:00