Construct a default optimizer */
| 121 | |
| 122 | /** Construct a default optimizer */ |
| 123 | Optimizer(int _type) : type("Default"), init_lr(0), lr(0) { |
| 124 | CHECK(_type == kAuto) << "Only kAuto can be used for initializing a default optimizer. " |
| 125 | << "Please use a float value if you want to specify the learning rate."; |
| 126 | } |
| 127 | |
| 128 | /** Construct a default optimizer with learning rate */ |
| 129 | Optimizer(float _lr = 1e-4) : type("Default"), init_lr(_lr), lr(_lr) {} |
no outgoing calls
no test coverage detected