code: @Override
      public long numParams() {
          int ret = 0;
          for (INDArray val : params.values())
              ret += val.length();
          return ret;
      }
docs: The number of parameters for the model
  
  @return the number of parameters for the model