!6553 [ModelZoo]Change gat seed
Merge pull request !6553 from zhanke/gat
This commit is contained in:
commit
50ee907cce
|
@ -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…
Reference in New Issue