darknet源码学习:forward_network_gpu
layer结构体中forward_gpu定义:
void(*forward_gpu) (struct layer, struct network_state);
不同网络层forward_gpu指向不同函数:
darknet源码解析:network结构体之n中,可以看出cfg文件中包含四种类型网络层:CONVOLUTIONAL,SOFTMAX,AVGPOOL,MAXPOOL。
darknet源码解析:forward_convolutional_layer_gpu
darknet源码解析:forward_maxpool_layer_gpu
darknet源码解析:forward_softmax_layer_gpu
darknet源码解析:forward_avgpool_layer_gpu