tidy

PyTorch/sparseconvnet/SCN/generic/Geometry/InputLayerRules.h

@@ -193,7 +193,6 @@ void blRules(SparseGrids<dimension> &SGs, RuleBook &rules, long *coords,
     rule.resize((maxActive + 1) * nActive);
 #pragma omp parallel for private(I)
     for (I = 0; I < batchSize; I++) {
-      std::cout << omp_get_num_threads() << "\n";
       auto &ors = outputRows[I];
       auto rr = &rule[SGs[I].ctr * (maxActive + 1)];
       for (auto &row : ors) {

PyTorch/sparseconvnet/networkInNetwork.py

@@ -19,7 +19,7 @@ class NetworkInNetworkFunction(Function):
             weight,
             bias):
         output_features = input_features.new()
-        ctx.save_forbackwards(input_features,
+        ctx.save_for_backward(input_features,
                               output_features,
                               weight,
                               bias)
@@ -67,8 +67,7 @@ class NetworkInNetwork(Module):
             0,
             std))
         if bias:
-            self.bias = Parameter(torch.Tensor(nOut).
-                                  _())
+            self.bias = Parameter(torch.Tensor(nOut).zero_())
         else:
             self.bias = None