ResNet working

This commit is contained in:
Davide Sapienza
2019-10-29 17:21:31 +01:00
parent e9ec582223
commit 02936fa928
5 changed files with 242 additions and 317 deletions
+5 -2
View File
@@ -7,9 +7,12 @@ namespace tk { namespace dnn {
Route::Route(Network *net, Layer **layers, int layers_n) : Layer(net) {
this->layers = layers;
this->layers_n = layers_n;
if(layers_n > MAX_INPUT_LAYERS)
FatalError("Route: MAX INPUT LAYERS overload");
for(int i=0; i<layers_n; i++)
this->layers[i] = layers[i];
//get dims
output_dim.l = 1;
output_dim.c = 0;