Compute current learning rate according to the schedule */
| 130 | |
| 131 | /** Compute current learning rate according to the schedule */ |
| 132 | void apply_schedule(int batch_id, int num_batch) { |
| 133 | lr = init_lr * schedule(batch_id, num_batch); |
| 134 | } |
| 135 | |
| 136 | /** Return information about the optimizer */ |
| 137 | std::string info() const { |