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,9 @@

 namespace tkDNN {

-LayerWgs::LayerWgs(Network *net, dataDim_t in_dim, 
-                int inputs, int outputs, int kh, int kw, int kl, 
-                const char* fname_weights, bool batchnorm) : Layer(net, in_dim) {
+LayerWgs::LayerWgs(Network *net, int inputs, int outputs, 
+                   int kh, int kw, int kl, 
+                   const char* fname_weights, bool batchnorm) : Layer(net) {

    this->inputs  = inputs;
    this->outputs = outputs;