change gcn early_stop epoch

4 years ago · 316448f4d4
parent 2a03fa79f9
commit 316448f4d4
1 changed files with 1 additions and 1 deletions
--- a/model_zoo/official/gnn/gcn/src/config.py
+++ b/model_zoo/official/gnn/gcn/src/config.py
@ -23,4 +23,4 @@ class ConfigGCN():
    hidden1 = 16
    dropout = 0.5
    weight_decay = 5e-4
-    early_stopping = 10
+    early_stopping = 50