From 54f88ddbb17774a72b74ec28279eb8e564d44c43 Mon Sep 17 00:00:00 2001
From: sys1874 <578417645@qq.com>
Date: Wed, 9 Sep 2020 09:51:01 +0800
Subject: [PATCH] Update model.py

---
 ogb_examples/nodeproppred/unimp/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ogb_examples/nodeproppred/unimp/model.py b/ogb_examples/nodeproppred/unimp/model.py
index c32d631..684e7aa 100644
--- a/ogb_examples/nodeproppred/unimp/model.py
+++ b/ogb_examples/nodeproppred/unimp/model.py
@@ -268,7 +268,7 @@ class Arxiv_label_embedding_model():
                                              hidden_size=self.hidden_size,
                                              num_heads=self.num_heads, 
                                              concat=True, 
-                                             layer_norm=True, relu=True)
+                                             layer_norm=True, relu=True, gate=True)
             if self.dropout > 0:
                 feature_batch = L.dropout(feature_batch, dropout_prob=self.dropout, 
                                      dropout_implementation='upscale_in_train') 
-- 
GitLab