| 1189 | |
| 1190 | protected: |
| 1191 | virtual void InitSolver(const SolverParameter& param) { |
| 1192 | const Dtype rms_decay = 0.95; |
| 1193 | SolverParameter new_param = param; |
| 1194 | new_param.set_rms_decay(rms_decay); |
| 1195 | this->solver_.reset(new RMSPropSolver<Dtype>(new_param)); |
| 1196 | } |
| 1197 | }; |
| 1198 | |
| 1199 | TYPED_TEST_CASE(RMSPropSolverTest, TestDtypesAndDevices); |