| 686 | update_rule(gscale, {&p->all_values, &p->all_grads, &vlp[idx].all_h}); |
| 687 | } |
| 688 | unsigned MomentumSGDTrainer::alloc_impl() { |
| 689 | allocate_shadow_parameters(*model, aux_allocated, vp); |
| 690 | return vp.size(); |
| 691 | } |
| 692 | unsigned MomentumSGDTrainer::alloc_lookup_impl() { |
| 693 | allocate_shadow_lookup_parameters(*model, aux_allocated_lookup, vlp); |
| 694 | return vlp.size(); |
nothing calls this directly
no test coverage detected