From a7fda0a4503f0c2db3f1c63a106020f562e97feb Mon Sep 17 00:00:00 2001
From: Somefive <somefive@foxmail.com>
Date: Mon, 22 Feb 2021 20:07:44 +0800
Subject: [PATCH] remove unnecessary cuda code

---
 cogdl/oag/bert_model.py | 4 ----
 1 file changed, 4 deletions(-)

diff --git a/cogdl/oag/bert_model.py b/cogdl/oag/bert_model.py
index 099e0a1f..02248f63 100644
--- 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