remove unnecessary cuda code

THUDM · Feb 22, 2021 · a7fda0a · a7fda0a
1 parent 92e5b87
commit a7fda0a
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/cogdl/oag/bert_model.py b/cogdl/oag/bert_model.py
@@ -413,11 +413,7 @@ def forward(self, hidden_states, masked_token_indexes):
                 hidden_states.view(-1, hidden_states.shape[-1]), 0,
                 masked_token_indexes)
 
-        torch.cuda.nvtx.range_push(
-            "decoder input.size() = {}, weight.size() = {}".format(
-                hidden_states.size(), self.decoder.weight.size()))
         hidden_states = self.decoder(hidden_states) + self.bias
-        torch.cuda.nvtx.range_pop()
         return hidden_states