Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

change gat seed

This commit is contained in:
zhanke 2020-09-19 14:08:54 +08:00
parent daf6739b22
commit ade0865d29
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
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.gat import GAT
from src.utils import LossAccuracyWrapper, TrainGAT from src.utils import LossAccuracyWrapper, TrainGAT
set_seed(1) set_seed(0)
def train(): def train():
"""Train GAT model.""" """Train GAT model."""