C++ (Cpp) layer::update_gpu Exemples

Exemple #1

Fichier : network_kernels.c Projet : samlong-yang/darknet

void update_layer(layer l, network net)
{
    int update_batch = net.batch*net.subdivisions;
    float rate = get_current_rate(net);
    l.t = get_current_batch(net);
    if(l.update_gpu){
        l.update_gpu(l, update_batch, rate*l.learning_rate_scale, net.momentum, net.decay);
    }
}