mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity
75 Commits 141 Branches 65 Tags
Commit Graph

2 Commits

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