Browse Source

!3283 change directory of GCN and GAT from model_zoo to model_zoo/official/gnn

Merge pull request !3283 from chentingting/change_gnn_dir2
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
abcee8e586
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      tests/st/gnn/gcn/test_gcn.py

+ 4
- 4
tests/st/gnn/gcn/test_gcn.py View File

@@ -17,10 +17,10 @@ import time
import pytest
import numpy as np
from mindspore import context
from model_zoo.gnn.gcn.src.gcn import GCN
from model_zoo.gnn.gcn.src.metrics import LossAccuracyWrapper, TrainNetWrapper
from model_zoo.gnn.gcn.src.config import ConfigGCN
from model_zoo.gnn.gcn.src.dataset import get_adj_features_labels, get_mask
from model_zoo.official.gnn.gcn.src.gcn import GCN
from model_zoo.official.gnn.gcn.src.metrics import LossAccuracyWrapper, TrainNetWrapper
from model_zoo.official.gnn.gcn.src.config import ConfigGCN
from model_zoo.official.gnn.gcn.src.dataset import get_adj_features_labels, get_mask


DATA_DIR = '/home/workspace/mindspore_dataset/cora/cora_mr/cora_mr'


Loading…
Cancel
Save