search

tusen-ai / simpledet

A Simple and Versatile Framework for Object Detection and Instance Recognition
Apache License 2.0
3.08k stars 488 forks source link

Optimizer/ Solver #234

Closed mistaleee closed 5 years ago

mistaleee commented 5 years ago

Hi all,

I've been searching for while to get to know which solver/optimizer (e.g. gradient descent, Adam etc. ) Trident is actually using.

Can anybody give me a hint how I can change the solver?

Thanks.

RogerChern commented 5 years ago

https://github.com/TuSimple/simpledet/blob/dd4eb159cf3934b949a89b745f00ddc62c0745c9/config/tridentnet_r50v1c4_c5_1x.py#L158-L160

mistaleee commented 5 years ago

Thanks. Which solvers would be accepted instead of sdg?

xchani commented 5 years ago

Check this https://mxnet.incubator.apache.org/api/python/docs/api/gluon-related/mxnet.optimizer.html . Remember to use lowercase of these names, i.e. sgd, adam, nag, etc.

mistaleee commented 5 years ago

Thanks @xchani