Browse Source

change gat seed

tags/v1.0.0
zhanke 5 years ago
parent
commit
ade0865d29
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      model_zoo/official/gnn/gat/train.py

+ 1
- 1
model_zoo/official/gnn/gat/train.py View File

@@ -26,7 +26,7 @@ from src.dataset import load_and_process
from src.gat import GAT
from src.utils import LossAccuracyWrapper, TrainGAT

set_seed(1)
set_seed(0)

def train():
"""Train GAT model."""


Loading…
Cancel
Save