better network model

2017-08-01 23:03:02 +02:00
parent 300b0af5dd
commit e8355cee67
22 changed files with 166 additions and 179 deletions
@@ -4,9 +4,8 @@

 namespace tkDNN {

-Dense::Dense(Network *net, dataDim_t in_dim, 
-    int out_ch, const char* fname_weights) : 
-    LayerWgs(net, in_dim, in_dim.tot(), out_ch, 1, 1, 1, fname_weights) {
+Dense::Dense(Network *net, int out_ch, const char* fname_weights) : 
+    LayerWgs(net, net->getOutputDim().tot(), out_ch, 1, 1, 1, fname_weights) {

    output_dim.n = 1;
    output_dim.c = out_ch;