dmlc--dgl
Chang Liu
885be1784b
* Refactor GCN example * Refactor GCN based on graphsage * Readme update * Minor update * update * Remove user-defined GCN implementation * README update * Update * Update CONTRIBUTORS.md * update task_example_test Co-authored-by: Xin Yao <xiny@nvidia.com>
36 行
615 B
Batchfile
36 行
615 B
Batchfile
@ECHO OFF
|
|
SETLOCAL EnableDelayedExpansion
|
|
|
|
SET GCN_EXAMPLE_DIR=.\examples\pytorch
|
|
|
|
IF x%1x==xx (
|
|
ECHO Must supply CPU or GPU
|
|
GOTO :FAIL
|
|
) ELSE IF x%1x==xcpux (
|
|
SET DEV=-1
|
|
) ELSE IF x%1x==xgpux (
|
|
SET DEV=0
|
|
SET CUDA_VISIBLE_DEVICES=0
|
|
) ELSE (
|
|
ECHO Must supply CPU or GPU
|
|
GOTO :FAIL
|
|
)
|
|
CALL workon %BUILD_TAG%
|
|
|
|
SET DGLBACKEND=pytorch
|
|
SET DGL_LIBRARY_PATH=!CD!\build
|
|
SET PYTHONPATH=!CD!\python;!PYTHONPATH!
|
|
SET DGL_DOWNLOAD_DIR=!CD!
|
|
|
|
PUSHD !GCN_EXAMPLE_DIR!
|
|
python pagerank.py || GOTO :FAIL
|
|
python gcn\train.py --dataset cora || GOTO :FAIL
|
|
POPD
|
|
ENDLOCAL
|
|
EXIT /B
|
|
|
|
:FAIL
|
|
ECHO Example test failed
|
|
ENDLOCAL
|
|
EXIT /B 1
|