cleanup

undertherain · May 21, 2020 · adab4b1 · adab4b1
1 parent c473cc6
commit adab4b1
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/benchmarker/modules/problems/bert/pytorch.py b/benchmarker/modules/problems/bert/pytorch.py
@@ -107,13 +107,9 @@ def forward(self, src, has_mask=True):
         src = self.encoder(src) * math.sqrt(self.ninp)
         src = self.pos_encoder(src)
         output = self.transformer_encoder(src, self.src_mask)
-        #print("bert last: ", output.shape)
         output = output[-1]
         output = self.decoder(output)
-        #print("decoded: ", output.shape)
-        #exit(0)
         # TODO: return softmax or cross_entropy depending on the mode
-        #output = output.
         return output
         # return F.log_softmax(output, dim=-1)