diff --git a/README.md b/README.md index 8bd2f638..876b9473 100644 --- a/README.md +++ b/README.md @@ -21,7 +21,7 @@ We summarize the contributions of CogDL as follows: ## โ News -- The new **v0.5.0-alpha0 pre-release** designs and implements a unified training loop for GNN. It introduces `DataWrapper` to help prepare the training/validation/test data and `ModelWrapper` to define the training/validation/test steps. +- The new **v0.5.0 release** designs and implements a unified training loop for GNN. It introduces `DataWrapper` to help prepare the training/validation/test data and `ModelWrapper` to define the training/validation/test steps. ๐ - The new **v0.4.1 release** adds the implementation of Deep GNNs and the recommendation task. It also supports new pipelines for generating embeddings and recommendation. Welcome to join our tutorial on KDD 2021 at 10:30 am - 12:00 am, Aug. 14th (Singapore Time). More details can be found in https://kdd2021graph.github.io/. ๐ diff --git a/README_CN.md b/README_CN.md index 73b894f3..39b398ea 100644 --- a/README_CN.md +++ b/README_CN.md @@ -21,7 +21,7 @@ CogDL็็นๆงๅ ๆฌ๏ผ ## โ ๆๆฐ -- ๆๆฐ็ **v0.5.0-alpha0 pre-release** ไธบๅพ็ฅ็ป็ฝ็ป็่ฎญ็ป่ฎพ่ฎกไบไธๅฅ็ปไธ็ๆต็จ. ่ฟไธช็ๆฌๅป้คไบๅๅ ็`Task`็ฑป๏ผๅผๅ ฅไบ`DataWrapper`ๆฅๅๅคtraining/validation/test่ฟ็จไธญๆ้็ๆฐๆฎ๏ผๅผๅ ฅไบ`ModelWrapper`ๆฅๅฎไนๆจกๅtraining/validation/test็ๆญฅ้ชค. +- ๆๆฐ็ **v0.5.0 release** ไธบๅพ็ฅ็ป็ฝ็ป็่ฎญ็ป่ฎพ่ฎกไบไธๅฅ็ปไธ็ๆต็จ. ่ฟไธช็ๆฌๅป้คไบๅๅ ็`Task`็ฑป๏ผๅผๅ ฅไบ`DataWrapper`ๆฅๅๅคtraining/validation/test่ฟ็จไธญๆ้็ๆฐๆฎ๏ผๅผๅ ฅไบ`ModelWrapper`ๆฅๅฎไนๆจกๅtraining/validation/test็ๆญฅ้ชค. ๐ - ๆๆฐ็ **v0.4.1 release** ๅขๅ ไบๆทฑๅฑGNN็ๅฎ็ฐๅๆจ่ไปปๅกใ่ฟไธช็ๆฌๅๆถๆไพไบๆฐ็ไธไบpipeline็จไบ็ดๆฅ่ทๅๅพ่กจ็คบๅๆญๅปบๆจ่ๅบ็จใๆฌข่ฟๅคงๅฎถๅๅ ๆไปฌๅจKDD 2021ไธ็tutorial๏ผๆถ้ดๆฏ8ๆ14ๅทไธๅ10:30 - 12:00๏ผๅไบฌๆถ้ด๏ผใ ๆดๅค็ๅ ๅฎนๅฏไปฅๆฅ็ https://kdd2021graph.github.io/. ๐ diff --git a/cogdl/__init__.py b/cogdl/__init__.py index 9018dea9..1407ccee 100644 --- a/cogdl/__init__.py +++ b/cogdl/__init__.py @@ -1,4 +1,4 @@ -__version__ = "0.5.0-alpha1" +__version__ = "0.5.0" from .experiments import experiment from .oag import oagbert diff --git a/docs/source/index.rst b/docs/source/index.rst index 5848584b..2481e698 100644 --- a/docs/source/index.rst +++ b/docs/source/index.rst @@ -17,7 +17,7 @@ We summarize the contributions of CogDL as follows: โ News ------------ -- The new **v0.5.0-alpha0 pre-release** designs and implements a unified training loop for GNN. It introduces `DataWrapper` to help prepare the training/validation/test data and `ModelWrapper` to define the training/validation/test steps. +- The new **v0.5.0 release** designs and implements a unified training loop for GNN. It introduces `DataWrapper` to help prepare the training/validation/test data and `ModelWrapper` to define the training/validation/test steps. - The new **v0.4.1 release** adds the implementation of Deep GNNs and the recommendation task. It also supports new pipelines for generating embeddings and recommendation. Welcome to join our tutorial on KDD 2021 at 10:30 am - 12:00 am, Aug. 14th (Singapore Time). More details can be found in https://kdd2021graph.github.io/. ๐ - The new **v0.4.0 release** refactors the data storage (from ``Data`` to ``Graph``) and provides more fast operators to speed up GNN training. It also includes many self-supervised learning methods on graphs. BTW, we are glad to announce that we will give a tutorial on KDD 2021 in August. Please see this `link <https://kdd2021graph.github.io/>`_ for more details. ๐ - The new **v0.3.0 release** provides a fast spmm operator to speed up GNN training. We also release the first version of `CogDL paper <https://arxiv.org/abs/2103.00959>`_ in arXiv. You can join `our slack <https://join.slack.com/t/cogdl/shared_invite/zt-b9b4a49j-2aMB035qZKxvjV4vqf0hEg>`_ for discussion. ๐๐๐ diff --git a/examples/notebooks/quickstart.ipynb b/examples/notebooks/quickstart.ipynb index 9bdec9d8..2774c40e 100644 --- a/examples/notebooks/quickstart.ipynb +++ b/examples/notebooks/quickstart.ipynb @@ -18,56 +18,6 @@ { "cell_type": "code", "execution_count": 1, - "metadata": { - "colab": { - "base_uri": "https://localhost:8080/" - }, - "id": "YOjTWjGz9CZo", - "outputId": "a78c7d89-9dfd-4146-a0f8-ad6dd1139f53" - }, - "outputs": [ - { - "name": "stdout", - "output_type": "stream", - "text": [ - "nvcc: NVIDIA (R) Cuda compiler driver\n", - "Copyright (c) 2005-2020 NVIDIA Corporation\n", - "Built on Wed_Jul_22_19:09:09_PDT_2020\n", - "Cuda compilation tools, release 11.0, V11.0.221\n", - "Build cuda_11.0_bu.TC445_37.28845127_0\n", - "Mon Jul 12 13:46:17 2021 \n", - "+-----------------------------------------------------------------------------+\n", - "| NVIDIA-SMI 470.42.01 Driver Version: 460.32.03 CUDA Version: 11.2 |\n", - "|-------------------------------+----------------------+----------------------+\n", - "| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |\n", - "| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |\n", - "| | | MIG M. |\n", - "|===============================+======================+======================|\n", - "| 0 Tesla T4 Off | 00000000:00:04.0 Off | 0 |\n", - "| N/A 66C P8 11W / 70W | 0MiB / 15109MiB | 0% Default |\n", - "| | | N/A |\n", - "+-------------------------------+----------------------+----------------------+\n", - " \n", - "+-----------------------------------------------------------------------------+\n", - "| Processes: |\n", - "| GPU GI CI PID Type Process name GPU Memory |\n", - "| ID ID Usage |\n", - "|=============================================================================|\n", - "| No running processes found |\n", - "+-----------------------------------------------------------------------------+\n" - ] - } - ], - "source": [ - "#@title see gpu card\n", - "\n", - "!nvcc -V\n", - "!nvidia-smi\n" - ] - }, - { - "cell_type": "code", - "execution_count": 2, "metadata": { "colab": { "base_uri": "https://localhost:8080/" @@ -80,153 +30,84 @@ "name": "stdout", "output_type": "stream", "text": [ - "Collecting cogdl\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/3e/c3/5590b87e06183fc85f7aad129b30c8ed4c0a0cca04857a9c2878686a804f/cogdl-0.4.0-py3-none-any.whl (324kB)\n", - "\r\u001b[K |โ | 10kB 13.7MB/s eta 0:00:01\r\u001b[K |โโ | 20kB 19.6MB/s eta 0:00:01\r\u001b[K |โโโ | 30kB 16.3MB/s eta 0:00:01\r\u001b[K |โโโโ | 40kB 14.7MB/s eta 0:00:01\r\u001b[K |โโโโโ | 51kB 8.6MB/s eta 0:00:01\r\u001b[K |โโโโโโ | 61kB 8.3MB/s eta 0:00:01\r\u001b[K |โโโโโโโ | 71kB 9.3MB/s eta 0:00:01\r\u001b[K |โโโโโโโโ | 81kB 9.7MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโ | 92kB 9.9MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโ | 102kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโ | 112kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโโ | 122kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโโโโ | 133kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโโโโโ | 143kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโโโโโโ | 153kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโโโโโโโ | 163kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโโโโโโโโ | 174kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโโโโโโโโโ | 184kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโโโโโโโโโโ | 194kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโโโโโโโโโโโ | 204kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโโโโโโโโโโโโ | 215kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโ | 225kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโ | 235kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโ | 245kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโ | 256kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโ | 266kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโ | 276kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโ | 286kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ | 296kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ | 307kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 317kB 8.2MB/s eta 0:00:01\r\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 327kB 8.2MB/s \n", - "\u001b[?25hCollecting grave\n", - " Downloading https://files.pythonhosted.org/packages/76/c4/930f7392dec82b39793d6f598cd151cc965ac59f87d87340a15379693700/grave-0.0.3-py3-none-any.whl\n", - "Requirement already satisfied: tabulate in /usr/local/lib/python3.7/dist-packages (from cogdl) (0.8.9)\n", - "Collecting pre-commit\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/b2/38/3e5475cbd6921cd8208a06c113384a88bb19ce0cda1155f33d4f19183de1/pre_commit-2.13.0-py2.py3-none-any.whl (190kB)\n", - "\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 194kB 35.5MB/s \n", - "\u001b[?25hCollecting optuna==2.4.0\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/59/b4/a1a80252cef3d8f5a0acdf6e678d6dc07e2e6964ee46d0453a2ae1af1ecb/optuna-2.4.0-py3-none-any.whl (282kB)\n", - "\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 286kB 32.0MB/s \n", - "\u001b[?25hRequirement already satisfied: torch in /usr/local/lib/python3.7/dist-packages (from cogdl) (1.9.0+cu102)\n", - "Collecting flake8\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/fc/80/35a0716e5d5101e643404dabd20f07f5528a21f3ef4032d31a49c913237b/flake8-3.9.2-py2.py3-none-any.whl (73kB)\n", - "\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 81kB 11.7MB/s \n", - "\u001b[?25hRequirement already satisfied: tqdm in /usr/local/lib/python3.7/dist-packages (from cogdl) (4.41.1)\n", - "Collecting ogb\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/d2/c5/20b1e4a5ff90ead06139ce1c2362474b97bb3a73ee0166eb37f2d3eb0dba/ogb-1.3.1-py3-none-any.whl (67kB)\n", - "\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 71kB 10.3MB/s \n", - "\u001b[?25hRequirement already satisfied: numba in /usr/local/lib/python3.7/dist-packages (from cogdl) (0.51.2)\n", - "Collecting sentencepiece\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/ac/aa/1437691b0c7c83086ebb79ce2da16e00bef024f24fec2a5161c35476f499/sentencepiece-0.1.96-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (1.2MB)\n", - "\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 1.2MB 33.6MB/s \n", - "\u001b[?25hRequirement already satisfied: networkx in /usr/local/lib/python3.7/dist-packages (from cogdl) (2.5.1)\n", - "Collecting texttable\n", - " Downloading https://files.pythonhosted.org/packages/06/f5/46201c428aebe0eecfa83df66bf3e6caa29659dbac5a56ddfd83cae0d4a4/texttable-1.6.3-py2.py3-none-any.whl\n", - "Collecting emoji\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/24/fa/b3368f41b95a286f8d300e323449ab4e86b85334c2e0b477e94422b8ed0f/emoji-1.2.0-py3-none-any.whl (131kB)\n", - "\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 133kB 53.2MB/s \n", - "\u001b[?25hRequirement already satisfied: scikit-learn in /usr/local/lib/python3.7/dist-packages (from cogdl) (0.22.2.post1)\n", - "Requirement already satisfied: scipy in /usr/local/lib/python3.7/dist-packages (from cogdl) (1.4.1)\n", - "Requirement already satisfied: numpy in /usr/local/lib/python3.7/dist-packages (from cogdl) (1.19.5)\n", - "Requirement already satisfied: gensim<4.0 in /usr/local/lib/python3.7/dist-packages (from cogdl) (3.6.0)\n", - "Requirement already satisfied: matplotlib in /usr/local/lib/python3.7/dist-packages (from cogdl) (3.2.2)\n", - "Collecting transformers\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/fd/1a/41c644c963249fd7f3836d926afa1e3f1cc234a1c40d80c5f03ad8f6f1b2/transformers-4.8.2-py3-none-any.whl (2.5MB)\n", - "\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 2.5MB 49.4MB/s \n", - "\u001b[?25hCollecting pyyaml>=5.1\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/7a/a5/393c087efdc78091afa2af9f1378762f9821c9c1d7a22c5753fb5ac5f97a/PyYAML-5.4.1-cp37-cp37m-manylinux1_x86_64.whl (636kB)\n", - "\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 645kB 54.0MB/s \n", - "\u001b[?25hCollecting cfgv>=2.0.0\n", - " Downloading https://files.pythonhosted.org/packages/49/54/83bf9b6ba673bf7d5ebe3846a5f6d3a53925cfd331aa29ec6b5b9c42a850/cfgv-3.3.0-py2.py3-none-any.whl\n", - "Requirement already satisfied: importlib-metadata; python_version < \"3.8\" in /usr/local/lib/python3.7/dist-packages (from pre-commit->cogdl) (4.6.0)\n", - "Collecting nodeenv>=0.11.1\n", - " Downloading https://files.pythonhosted.org/packages/54/73/56c89b343befb9c63e8117294d265458f0ff726fa2abcdc6bb5ec5e66a1a/nodeenv-1.6.0-py2.py3-none-any.whl\n", - "Collecting identify>=1.0.0\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/c7/b7/0f4009c9e23f7017be400e06dc0251671d64ba4fddaab82c98210d649a15/identify-2.2.11-py2.py3-none-any.whl (98kB)\n", - "\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 102kB 14.7MB/s \n", - "\u001b[?25hCollecting virtualenv>=20.0.8\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/03/08/f819421002e85a71d58368f7bffbe0b1921325e0e8ca7857cb5fb0e1f7c1/virtualenv-20.4.7-py2.py3-none-any.whl (7.2MB)\n", - "\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 7.2MB 51.2MB/s \n", - "\u001b[?25hRequirement already satisfied: toml in /usr/local/lib/python3.7/dist-packages (from pre-commit->cogdl) (0.10.2)\n", - "Collecting cmaes>=0.6.0\n", - " Downloading https://files.pythonhosted.org/packages/01/1f/43b01223a0366171f474320c6e966c39a11587287f098a5f09809b45e05f/cmaes-0.8.2-py3-none-any.whl\n", - "Collecting cliff\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/87/11/aea1cacbd4cf8262809c4d6f95dcb3f2802594de1f51c5bd454d69bf15c5/cliff-3.8.0-py3-none-any.whl (80kB)\n", - "\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 81kB 12.9MB/s \n", - "\u001b[?25hCollecting colorlog\n", - " Downloading https://files.pythonhosted.org/packages/32/e6/e9ddc6fa1104fda718338b341e4b3dc31cd8039ab29e52fc73b508515361/colorlog-5.0.1-py2.py3-none-any.whl\n", - "Requirement already satisfied: sqlalchemy>=1.1.0 in /usr/local/lib/python3.7/dist-packages (from optuna==2.4.0->cogdl) (1.4.20)\n", - "Requirement already satisfied: joblib in /usr/local/lib/python3.7/dist-packages (from optuna==2.4.0->cogdl) (1.0.1)\n", - "Collecting alembic\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/d5/80/ef186e599a57d0e4cb78fc76e0bfc2e6953fa9716b2a5cf2de0117ed8eb5/alembic-1.6.5-py2.py3-none-any.whl (164kB)\n", - "\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 174kB 58.4MB/s \n", - "\u001b[?25hRequirement already satisfied: packaging>=20.0 in /usr/local/lib/python3.7/dist-packages (from optuna==2.4.0->cogdl) (20.9)\n", - "Requirement already satisfied: typing-extensions in /usr/local/lib/python3.7/dist-packages (from torch->cogdl) (3.7.4.3)\n", - "Collecting pyflakes<2.4.0,>=2.3.0\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/6c/11/2a745612f1d3cbbd9c69ba14b1b43a35a2f5c3c81cd0124508c52c64307f/pyflakes-2.3.1-py2.py3-none-any.whl (68kB)\n", - "\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 71kB 11.6MB/s \n", - "\u001b[?25hCollecting mccabe<0.7.0,>=0.6.0\n", - " Downloading https://files.pythonhosted.org/packages/87/89/479dc97e18549e21354893e4ee4ef36db1d237534982482c3681ee6e7b57/mccabe-0.6.1-py2.py3-none-any.whl\n", - "Collecting pycodestyle<2.8.0,>=2.7.0\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/de/cc/227251b1471f129bc35e966bb0fceb005969023926d744139642d847b7ae/pycodestyle-2.7.0-py2.py3-none-any.whl (41kB)\n", - "\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 51kB 9.3MB/s \n", - "\u001b[?25hRequirement already satisfied: pandas>=0.24.0 in /usr/local/lib/python3.7/dist-packages (from ogb->cogdl) (1.1.5)\n", - "Requirement already satisfied: urllib3>=1.24.0 in /usr/local/lib/python3.7/dist-packages (from ogb->cogdl) (1.24.3)\n", - "Collecting outdated>=0.2.0\n", - " Downloading https://files.pythonhosted.org/packages/fd/f6/95588d496e518355c33b389222c99069b1c6f2c046be64f400072fdc7cda/outdated-0.2.1-py3-none-any.whl\n", - "Requirement already satisfied: six>=1.12.0 in /usr/local/lib/python3.7/dist-packages (from ogb->cogdl) (1.15.0)\n", - "Requirement already satisfied: setuptools in /usr/local/lib/python3.7/dist-packages (from numba->cogdl) (57.0.0)\n", - "Requirement already satisfied: llvmlite<0.35,>=0.34.0.dev0 in /usr/local/lib/python3.7/dist-packages (from numba->cogdl) (0.34.0)\n", - "Requirement already satisfied: decorator<5,>=4.3 in /usr/local/lib/python3.7/dist-packages (from networkx->cogdl) (4.4.2)\n", - "Requirement already satisfied: smart-open>=1.2.1 in /usr/local/lib/python3.7/dist-packages (from gensim<4.0->cogdl) (5.1.0)\n", - "Requirement already satisfied: kiwisolver>=1.0.1 in /usr/local/lib/python3.7/dist-packages (from matplotlib->cogdl) (1.3.1)\n", - "Requirement already satisfied: python-dateutil>=2.1 in /usr/local/lib/python3.7/dist-packages (from matplotlib->cogdl) (2.8.1)\n", - "Requirement already satisfied: cycler>=0.10 in /usr/local/lib/python3.7/dist-packages (from matplotlib->cogdl) (0.10.0)\n", - "Requirement already satisfied: pyparsing!=2.0.4,!=2.1.2,!=2.1.6,>=2.0.1 in /usr/local/lib/python3.7/dist-packages (from matplotlib->cogdl) (2.4.7)\n", - "Collecting tokenizers<0.11,>=0.10.1\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/d4/e2/df3543e8ffdab68f5acc73f613de9c2b155ac47f162e725dcac87c521c11/tokenizers-0.10.3-cp37-cp37m-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl (3.3MB)\n", - "\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 3.3MB 40.5MB/s \n", - "\u001b[?25hCollecting sacremoses\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/75/ee/67241dc87f266093c533a2d4d3d69438e57d7a90abb216fa076e7d475d4a/sacremoses-0.0.45-py3-none-any.whl (895kB)\n", - "\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 901kB 35.9MB/s \n", - "\u001b[?25hRequirement already satisfied: requests in /usr/local/lib/python3.7/dist-packages (from transformers->cogdl) (2.23.0)\n", - "Requirement already satisfied: regex!=2019.12.17 in /usr/local/lib/python3.7/dist-packages (from transformers->cogdl) (2019.12.20)\n", - "Requirement already satisfied: filelock in /usr/local/lib/python3.7/dist-packages (from transformers->cogdl) (3.0.12)\n", - "Collecting huggingface-hub==0.0.12\n", - " Downloading https://files.pythonhosted.org/packages/2f/ee/97e253668fda9b17e968b3f97b2f8e53aa0127e8807d24a547687423fe0b/huggingface_hub-0.0.12-py3-none-any.whl\n", - "Requirement already satisfied: zipp>=0.5 in /usr/local/lib/python3.7/dist-packages (from importlib-metadata; python_version < \"3.8\"->pre-commit->cogdl) (3.4.1)\n", - "Collecting distlib<1,>=0.3.1\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/87/26/f6a23dd3e578132cf924e0dd5d4e055af0cd4ab43e2a9f10b7568bfb39d9/distlib-0.3.2-py2.py3-none-any.whl (338kB)\n", - "\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 348kB 50.7MB/s \n", - "\u001b[?25hRequirement already satisfied: appdirs<2,>=1.4.3 in /usr/local/lib/python3.7/dist-packages (from virtualenv>=20.0.8->pre-commit->cogdl) (1.4.4)\n", - "Collecting cmd2>=1.0.0\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/44/ca/d407811641ec1d8bd8a38ee3165d73aa44776d7700436bd4d4a6606f2736/cmd2-2.1.2-py3-none-any.whl (141kB)\n", - "\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 143kB 57.0MB/s \n", - "\u001b[?25hCollecting pbr!=2.1.0,>=2.0.0\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/18/e0/1d4702dd81121d04a477c272d47ee5b6bc970d1a0990b11befa275c55cf2/pbr-5.6.0-py2.py3-none-any.whl (111kB)\n", - "\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 112kB 61.2MB/s \n", - "\u001b[?25hCollecting stevedore>=2.0.1\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/d4/49/b602307aeac3df3384ff1fcd05da9c0376c622a6c48bb5325f28ab165b57/stevedore-3.3.0-py3-none-any.whl (49kB)\n", - "\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 51kB 9.4MB/s \n", - "\u001b[?25hRequirement already satisfied: PrettyTable>=0.7.2 in /usr/local/lib/python3.7/dist-packages (from cliff->optuna==2.4.0->cogdl) (2.1.0)\n", - "Requirement already satisfied: greenlet!=0.4.17; python_version >= \"3\" in /usr/local/lib/python3.7/dist-packages (from sqlalchemy>=1.1.0->optuna==2.4.0->cogdl) (1.1.0)\n", - "Collecting python-editor>=0.3\n", - " Downloading https://files.pythonhosted.org/packages/c6/d3/201fc3abe391bbae6606e6f1d598c15d367033332bd54352b12f35513717/python_editor-1.0.4-py3-none-any.whl\n", - "Collecting Mako\n", - "\u001b[?25l Downloading https://files.pythonhosted.org/packages/f3/54/dbc07fbb20865d3b78fdb7cf7fa713e2cba4f87f71100074ef2dc9f9d1f7/Mako-1.1.4-py2.py3-none-any.whl (75kB)\n", - "\u001b[K |โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ| 81kB 12.5MB/s \n", - "\u001b[?25hRequirement already satisfied: pytz>=2017.2 in /usr/local/lib/python3.7/dist-packages (from pandas>=0.24.0->ogb->cogdl) (2018.9)\n", - "Collecting littleutils\n", - " Downloading https://files.pythonhosted.org/packages/4e/b1/bb4e06f010947d67349f863b6a2ad71577f85590180a935f60543f622652/littleutils-0.2.2.tar.gz\n", - "Requirement already satisfied: click in /usr/local/lib/python3.7/dist-packages (from sacremoses->transformers->cogdl) (7.1.2)\n", - "Requirement already satisfied: chardet<4,>=3.0.2 in /usr/local/lib/python3.7/dist-packages (from requests->transformers->cogdl) (3.0.4)\n", - "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.7/dist-packages (from requests->transformers->cogdl) (2021.5.30)\n", - "Requirement already satisfied: idna<3,>=2.5 in /usr/local/lib/python3.7/dist-packages (from requests->transformers->cogdl) (2.10)\n", - "Collecting pyperclip>=1.6\n", - " Downloading https://files.pythonhosted.org/packages/a7/2c/4c64579f847bd5d539803c8b909e54ba087a79d01bb3aba433a95879a6c5/pyperclip-1.8.2.tar.gz\n", - "Requirement already satisfied: wcwidth>=0.1.7 in /usr/local/lib/python3.7/dist-packages (from cmd2>=1.0.0->cliff->optuna==2.4.0->cogdl) (0.2.5)\n", - "Requirement already satisfied: attrs>=16.3.0 in /usr/local/lib/python3.7/dist-packages (from cmd2>=1.0.0->cliff->optuna==2.4.0->cogdl) (21.2.0)\n", - "Collecting colorama>=0.3.7\n", - " Downloading https://files.pythonhosted.org/packages/44/98/5b86278fbbf250d239ae0ecb724f8572af1c91f4a11edf4d36a206189440/colorama-0.4.4-py2.py3-none-any.whl\n", - "Requirement already satisfied: MarkupSafe>=0.9.2 in /usr/local/lib/python3.7/dist-packages (from Mako->alembic->optuna==2.4.0->cogdl) (2.0.1)\n", - "Building wheels for collected packages: littleutils, pyperclip\n", - " Building wheel for littleutils (setup.py) ... \u001b[?25l\u001b[?25hdone\n", - " Created wheel for littleutils: filename=littleutils-0.2.2-cp37-none-any.whl size=7050 sha256=e48bf89e06adb807a6e7ce06dfe43126822dd57078e780d45b9c460812275f27\n", - " Stored in directory: /root/.cache/pip/wheels/53/16/9f/ac67d15c40243754fd73f620e1b9b6dedc20492ecc19a2bae1\n", - " Building wheel for pyperclip (setup.py) ... \u001b[?25l\u001b[?25hdone\n", - " Created wheel for pyperclip: filename=pyperclip-1.8.2-cp37-none-any.whl size=11136 sha256=32ada558a3e9bde50292b0b2ba91b2f4319fa5bcb850ad73d2c7a2e52ecb8ede\n", - " Stored in directory: /root/.cache/pip/wheels/25/af/b8/3407109267803f4015e1ee2ff23be0c8c19ce4008665931ee1\n", - "Successfully built littleutils pyperclip\n", - "Installing collected packages: grave, pyyaml, cfgv, nodeenv, identify, distlib, virtualenv, pre-commit, cmaes, pyperclip, colorama, cmd2, pbr, stevedore, cliff, colorlog, python-editor, Mako, alembic, optuna, pyflakes, mccabe, pycodestyle, flake8, littleutils, outdated, ogb, sentencepiece, texttable, emoji, tokenizers, sacremoses, huggingface-hub, transformers, cogdl\n", - " Found existing installation: PyYAML 3.13\n", - " Uninstalling PyYAML-3.13:\n", - " Successfully uninstalled PyYAML-3.13\n", - "Successfully installed Mako-1.1.4 alembic-1.6.5 cfgv-3.3.0 cliff-3.8.0 cmaes-0.8.2 cmd2-2.1.2 cogdl-0.4.0 colorama-0.4.4 colorlog-5.0.1 distlib-0.3.2 emoji-1.2.0 flake8-3.9.2 grave-0.0.3 huggingface-hub-0.0.12 identify-2.2.11 littleutils-0.2.2 mccabe-0.6.1 nodeenv-1.6.0 ogb-1.3.1 optuna-2.4.0 outdated-0.2.1 pbr-5.6.0 pre-commit-2.13.0 pycodestyle-2.7.0 pyflakes-2.3.1 pyperclip-1.8.2 python-editor-1.0.4 pyyaml-5.4.1 sacremoses-0.0.45 sentencepiece-0.1.96 stevedore-3.3.0 texttable-1.6.3 tokenizers-0.10.3 transformers-4.8.2 virtualenv-20.4.7\n" + "Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple\n", + "Requirement already satisfied: cogdl in /home/yukuo/cogdl (0.5.0a1)\n", + "Requirement already satisfied: torch in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cogdl) (1.8.1)\n", + "Requirement already satisfied: networkx in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cogdl) (2.5.1)\n", + "Requirement already satisfied: matplotlib in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cogdl) (3.4.2)\n", + "Requirement already satisfied: tqdm in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cogdl) (4.62.2)\n", + "Requirement already satisfied: numpy<1.21,>=1.17 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cogdl) (1.20.1)\n", + "Requirement already satisfied: scipy in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cogdl) (1.6.3)\n", + "Requirement already satisfied: gensim<4.0 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cogdl) (3.8.0)\n", + "Requirement already satisfied: grave in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cogdl) (0.0.3)\n", + "Requirement already satisfied: scikit_learn in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cogdl) (0.24.2)\n", + "Requirement already satisfied: tabulate in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cogdl) (0.8.9)\n", + "Requirement already satisfied: optuna==2.4.0 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cogdl) (2.4.0)\n", + "Requirement already satisfied: ogb in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cogdl) (1.3.2)\n", + "Requirement already satisfied: pre-commit in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cogdl) (2.12.1)\n", + "Requirement already satisfied: flake8 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cogdl) (3.9.2)\n", + "Requirement already satisfied: numba in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cogdl) (0.53.1)\n", + "Requirement already satisfied: ninja in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cogdl) (1.10.2.2)\n", + "Requirement already satisfied: transformers in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cogdl) (4.5.1)\n", + "Requirement already satisfied: sentencepiece in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cogdl) (0.1.95)\n", + "Requirement already satisfied: alembic in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from optuna==2.4.0->cogdl) (1.6.2)\n", + "Requirement already satisfied: packaging>=20.0 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from optuna==2.4.0->cogdl) (20.9)\n", + "Requirement already satisfied: colorlog in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from optuna==2.4.0->cogdl) (5.0.1)\n", + "Requirement already satisfied: cmaes>=0.6.0 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from optuna==2.4.0->cogdl) (0.8.2)\n", + "Requirement already satisfied: joblib in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from optuna==2.4.0->cogdl) (1.0.1)\n", + "Requirement already satisfied: cliff in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from optuna==2.4.0->cogdl) (3.7.0)\n", + "Requirement already satisfied: sqlalchemy>=1.1.0 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from optuna==2.4.0->cogdl) (1.4.15)\n", + "Requirement already satisfied: smart-open>=1.7.0 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from gensim<4.0->cogdl) (5.0.0)\n", + "Requirement already satisfied: six>=1.5.0 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from gensim<4.0->cogdl) (1.15.0)\n", + "Requirement already satisfied: pyparsing>=2.0.2 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from packaging>=20.0->optuna==2.4.0->cogdl) (2.4.7)\n", + "Requirement already satisfied: greenlet!=0.4.17 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from sqlalchemy>=1.1.0->optuna==2.4.0->cogdl) (1.1.0)\n", + "Requirement already satisfied: importlib-metadata in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from sqlalchemy>=1.1.0->optuna==2.4.0->cogdl) (4.0.1)\n", + "Requirement already satisfied: python-dateutil in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from alembic->optuna==2.4.0->cogdl) (2.8.1)\n", + "Requirement already satisfied: python-editor>=0.3 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from alembic->optuna==2.4.0->cogdl) (1.0.4)\n", + "Requirement already satisfied: Mako in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from alembic->optuna==2.4.0->cogdl) (1.1.4)\n", + "Requirement already satisfied: PrettyTable>=0.7.2 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cliff->optuna==2.4.0->cogdl) (2.1.0)\n", + "Requirement already satisfied: stevedore>=2.0.1 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cliff->optuna==2.4.0->cogdl) (3.3.0)\n", + "Requirement already satisfied: cmd2>=1.0.0 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cliff->optuna==2.4.0->cogdl) (1.5.0)\n", + "Requirement already satisfied: pbr!=2.1.0,>=2.0.0 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cliff->optuna==2.4.0->cogdl) (5.6.0)\n", + "Requirement already satisfied: PyYAML>=3.12 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cliff->optuna==2.4.0->cogdl) (5.4.1)\n", + "Requirement already satisfied: wcwidth>=0.1.7 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cmd2>=1.0.0->cliff->optuna==2.4.0->cogdl) (0.2.5)\n", + "Requirement already satisfied: attrs>=16.3.0 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cmd2>=1.0.0->cliff->optuna==2.4.0->cogdl) (21.2.0)\n", + "Requirement already satisfied: colorama>=0.3.7 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cmd2>=1.0.0->cliff->optuna==2.4.0->cogdl) (0.4.4)\n", + "Requirement already satisfied: pyperclip>=1.6 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from cmd2>=1.0.0->cliff->optuna==2.4.0->cogdl) (1.8.2)\n", + "Requirement already satisfied: zipp>=0.5 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from importlib-metadata->sqlalchemy>=1.1.0->optuna==2.4.0->cogdl) (3.4.1)\n", + "Requirement already satisfied: typing-extensions>=3.6.4 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from importlib-metadata->sqlalchemy>=1.1.0->optuna==2.4.0->cogdl) (3.7.4.3)\n", + "Requirement already satisfied: pycodestyle<2.8.0,>=2.7.0 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from flake8->cogdl) (2.7.0)\n", + "Requirement already satisfied: pyflakes<2.4.0,>=2.3.0 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from flake8->cogdl) (2.3.1)\n", + "Requirement already satisfied: mccabe<0.7.0,>=0.6.0 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from flake8->cogdl) (0.6.1)\n", + "Requirement already satisfied: MarkupSafe>=0.9.2 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from Mako->alembic->optuna==2.4.0->cogdl) (2.0.1)\n", + "Requirement already satisfied: cycler>=0.10 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from matplotlib->cogdl) (0.10.0)\n", + "Requirement already satisfied: pillow>=6.2.0 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from matplotlib->cogdl) (8.2.0)\n", + "Requirement already satisfied: kiwisolver>=1.0.1 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from matplotlib->cogdl) (1.3.1)\n", + "Requirement already satisfied: decorator<5,>=4.3 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from networkx->cogdl) (4.4.2)\n", + "Requirement already satisfied: setuptools in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from numba->cogdl) (57.4.0)\n", + "Requirement already satisfied: llvmlite<0.37,>=0.36.0rc1 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from numba->cogdl) (0.36.0)\n", + "Requirement already satisfied: pandas>=0.24.0 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from ogb->cogdl) (1.2.4)\n", + "Requirement already satisfied: outdated>=0.2.0 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from ogb->cogdl) (0.2.1)\n", + "Requirement already satisfied: urllib3>=1.24.0 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from ogb->cogdl) (1.26.4)\n", + "Requirement already satisfied: requests in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from outdated>=0.2.0->ogb->cogdl) (2.25.1)\n", + "Requirement already satisfied: littleutils in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from outdated>=0.2.0->ogb->cogdl) (0.2.2)\n", + "Requirement already satisfied: pytz>=2017.3 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from pandas>=0.24.0->ogb->cogdl) (2021.1)\n", + "Requirement already satisfied: threadpoolctl>=2.0.0 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from scikit_learn->cogdl) (2.1.0)\n", + "Requirement already satisfied: cfgv>=2.0.0 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from pre-commit->cogdl) (3.2.0)\n", + "Requirement already satisfied: toml in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from pre-commit->cogdl) (0.10.2)\n", + "Requirement already satisfied: nodeenv>=0.11.1 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from pre-commit->cogdl) (1.6.0)\n", + "Requirement already satisfied: virtualenv>=20.0.8 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from pre-commit->cogdl) (20.4.6)\n", + "Requirement already satisfied: identify>=1.0.0 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from pre-commit->cogdl) (2.2.4)\n", + "Requirement already satisfied: filelock<4,>=3.0.0 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from virtualenv>=20.0.8->pre-commit->cogdl) (3.0.12)\n", + "Requirement already satisfied: distlib<1,>=0.3.1 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from virtualenv>=20.0.8->pre-commit->cogdl) (0.3.1)\n", + "Requirement already satisfied: appdirs<2,>=1.4.3 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from virtualenv>=20.0.8->pre-commit->cogdl) (1.4.4)\n", + "Requirement already satisfied: idna<3,>=2.5 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from requests->outdated>=0.2.0->ogb->cogdl) (2.10)\n", + "Requirement already satisfied: certifi>=2017.4.17 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from requests->outdated>=0.2.0->ogb->cogdl) (2020.12.5)\n", + "Requirement already satisfied: chardet<5,>=3.0.2 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from requests->outdated>=0.2.0->ogb->cogdl) (4.0.0)\n", + "Requirement already satisfied: tokenizers<0.11,>=0.10.1 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from transformers->cogdl) (0.10.2)\n", + "Requirement already satisfied: sacremoses in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from transformers->cogdl) (0.0.45)\n", + "Requirement already satisfied: regex!=2019.12.17 in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from transformers->cogdl) (2021.4.4)\n", + "Requirement already satisfied: click in /home/yukuo/anaconda3/envs/cu102/lib/python3.7/site-packages (from sacremoses->transformers->cogdl) (7.1.2)\n" ] } ], @@ -238,7 +119,7 @@ }, { "cell_type": "code", - "execution_count": 3, + "execution_count": 2, "metadata": { "colab": { "base_uri": "https://localhost:8080/" @@ -251,49 +132,41 @@ "name": "stdout", "output_type": "stream", "text": [ - "Failed to load C version of sampling, use python version instead.\n", - "Namespace(activation='relu', checkpoint=None, cpu=False, dataset='cora', device_id=[0], dropout=0.5, fast_spmm=False, hidden_size=64, inference=False, lr=0.01, max_epoch=500, missing_rate=0, model='gcn', norm=None, num_classes=None, num_features=None, num_layers=2, patience=100, residual=False, save_dir='.', save_model=None, seed=1, task='node_classification', trainer=None, use_best_config=False, weight_decay=0.0005)\n", - "Downloading https://cloud.tsinghua.edu.cn/d/6808093f7f8042bfa1f0/files/?p=%2Fcora.zip&dl=1\n", - "unpacking cora.zip\n", - "Processing...\n", - "Done!\n" + "Namespace(activation='relu', actnn=False, checkpoint_path='./checkpoints/model.pt', cpu=False, cpu_inference=False, dataset=['cora'], devices=[0], distributed=False, dropout=0.5, dw='node_classification_dw', epochs=500, eval_step=1, hidden_size=64, load_emb_path=None, local_rank=0, log_path='.', logger=None, lr=0.01, master_addr='localhost', master_port=13425, max_epoch=None, model=['gcn'], mw='node_classification_mw', n_trials=3, n_warmup_steps=0, no_test=False, norm=None, nstage=1, num_classes=None, num_features=None, num_layers=2, patience=100, progress_bar='epoch', project='cogdl-exp', residual=False, resume_training=False, rp_ratio=1, save_emb_path=None, seed=[1], split=[0], unsup=False, use_best_config=False, weight_decay=0)\n", + " \n", + "|----------------------------------------------------------------------------------------|\n", + " *** Running (`cora`, `gcn`, `node_classification_dw`, `node_classification_mw`)\n", + "|----------------------------------------------------------------------------------------|\n", + "Model Parameters: 92231\n" ] }, { "name": "stderr", "output_type": "stream", "text": [ - "Epoch: 455, Train: 1.0000, Val: 0.7920, ValLoss: 0.7300: 89%|โโโโโโโโโ | 446/500 [00:03<00:00, 120.41it/s]" + "Epoch: 146, train_loss: 0.0109, val_acc: 0.7860: 29%|โโโ | 147/500 [00:02<00:05, 66.46it/s] \n" ] }, { "name": "stdout", "output_type": "stream", "text": [ - "Valid accurracy = 0.7940\n", - "Test accuracy = 0.8100\n", - "| Variant | Acc | ValAcc |\n", + "Saving 47-th model to ./checkpoints/model.pt ...\n", + "Loading model from ./checkpoints/model.pt ...\n", + "{'test_acc': 0.817, 'val_acc': 0.798}\n", + "| Variant | test_acc | val_acc |\n", "|-----------------|---------------|---------------|\n", - "| ('cora', 'gcn') | 0.8100ยฑ0.0000 | 0.7940ยฑ0.0000 |\n" - ] - }, - { - "name": "stderr", - "output_type": "stream", - "text": [ - "\n" + "| ('cora', 'gcn') | 0.8170ยฑ0.0000 | 0.7980ยฑ0.0000 |\n" ] }, { "data": { "text/plain": [ - "defaultdict(list, {('cora', 'gcn'): [{'Acc': 0.81, 'ValAcc': 0.794}]})" + "defaultdict(list, {('cora', 'gcn'): [{'test_acc': 0.817, 'val_acc': 0.798}]})" ] }, - "execution_count": 3, - "metadata": { - "tags": [] - }, + "execution_count": 2, + "metadata": {}, "output_type": "execute_result" } ], @@ -466,14 +339,14 @@ ], "source": [ "# automl usage\n", - "def func_search(trial):\n", + "def search_space(trial):\n", " return {\n", " \"lr\": trial.suggest_categorical(\"lr\", [1e-3, 5e-3, 1e-2]),\n", " \"hidden_size\": trial.suggest_categorical(\"hidden_size\", [32, 64, 128]),\n", " \"dropout\": trial.suggest_uniform(\"dropout\", 0.5, 0.8),\n", " }\n", "\n", - "experiment(dataset=\"cora\", model=\"gcn\", seed=[1, 2], func_search=func_search)" + "experiment(dataset=\"cora\", model=\"gcn\", seed=[1, 2], search_space=search_space)" ] } ], @@ -486,6 +359,18 @@ "kernelspec": { "display_name": "Python 3", "name": "python3" + }, + "language_info": { + "codemirror_mode": { + "name": "ipython", + "version": 3 + }, + "file_extension": ".py", + "mimetype": "text/x-python", + "name": "python", + "nbconvert_exporter": "python", + "pygments_lexer": "ipython3", + "version": "3.7.10" } }, "nbformat": 4, diff --git a/third_party/dgNN b/third_party/dgNN index fc12618e..e1cbdc7c 160000 --- a/third_party/dgNN +++ b/third_party/dgNN @@ -1 +1 @@ -Subproject commit fc12618ed598a303fc824e660303d46b358d4849 +Subproject commit e1cbdc7c6c02c6ff0865837f6a2d02b24199108f diff --git a/third_party/fastmoe b/third_party/fastmoe index 3397bc19..b652e8d8 160000 --- a/third_party/fastmoe +++ b/third_party/fastmoe @@ -1 +1 @@ -Subproject commit 3397bc19ebe1568850980587a156abd79a5c5465 +Subproject commit b652e8d88bbe82171fa28a479f21ad1263fb2e1c