Added STNet-prepended TransCrowd GAP

model/stn.py

@@ -1,4 +1,6 @@
 # stole from pytorch tutorial
+# "Great artists steal" -- they say,
+# but thieves also steal so you know :P
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
@@ -36,7 +38,7 @@ class STNet(nn.Module):
 
         # (3.2) Regressor for the 3 * 2 affine matrix
         self.fc_loc = nn.Sequential(
-            # XXX: Should
+            # XXX: Dimensionality reduction across channels or not?
             nn.Linear(np.prod(self._loc_net_out_shape), 32),
             nn.ReLU(True),
             nn.Linear(32, 3 * 2)