SubdivNet.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "SubdivNet.ipynb",
      "provenance": [],
      "machine_shape": "hm",
      "mount_file_id": "1YMzovdO35scWfyZ_JoWdtHoVkHVbPRdn",
      "authorship_tag": "ABX9TyMZom1p7b9oziyrCzh1i1fg",
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    },
    "accelerator": "GPU"
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/DIPTE/Deep-Geometry-Learning-on-Colab/blob/master/SubdivNet.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "xvOpVpqfJlxn",
        "outputId": "10eb3fb7-3b8a-4f01-aa56-14ce5b247102"
      },
      "source": [
        "!ls"
      ],
      "execution_count": 9,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "airplane.obj\t maps\t\t   scripts     train_cls.py\n",
            "datagen_maps.py  README.md\t   subdivnet   train_seg.py\n",
            "LICENSE\t\t requirements.txt  teaser.jpg\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "H1NkZXc7J0Es"
      },
      "source": [
        "from google.colab import drive\n",
        "drive.mount('/content/drive')"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "PYuVshddJ78A"
      },
      "source": [
        "!nvcc -V"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "5udI1YFeJ-LF"
      },
      "source": [
        "import os\n",
        "os.chdir('/content/drive/My Drive')#切换工作目录"
      ],
      "execution_count": 4,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "SzByW6YEKBKd",
        "outputId": "83e85374-e93e-4e31-ce71-7bd73562f860"
      },
      "source": [
        "!git clone https://github.com/lzhengning/SubdivNet.git"
      ],
      "execution_count": 6,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "Cloning into 'SubdivNet'...\n",
            "remote: Enumerating objects: 124, done.\u001b[K\n",
            "remote: Counting objects: 100% (124/124), done.\u001b[K\n",
            "remote: Compressing objects: 100% (89/89), done.\u001b[K\n",
            "remote: Total 124 (delta 56), reused 89 (delta 33), pack-reused 0\u001b[K\n",
            "Receiving objects: 100% (124/124), 712.63 KiB | 9.50 MiB/s, done.\n",
            "Resolving deltas: 100% (56/56), done.\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "_LIpzlgfKImf"
      },
      "source": [
        "import os\n",
        "os.chdir('/content/drive/My Drive/SubdivNet')#切换工作目录"
      ],
      "execution_count": 8,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "pDNqbVXKKPyH",
        "outputId": "92d6cbf6-a8dc-4e66-cc48-6be5a94531bf"
      },
      "source": [
        "!pip install -r requirements.txt"
      ],
      "execution_count": 10,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "Collecting jittor\n",
            "\u001b[?25l  Downloading https://files.pythonhosted.org/packages/8f/6e/3a2a652e92a320cf7c52f6d380b9c91ac11cb6f568bd7d24a2fc1387e1cf/jittor-1.2.3.55.tar.gz (1.1MB)\n",
            "\u001b[K     |████████████████████████████████| 1.1MB 12.1MB/s \n",
            "\u001b[?25hRequirement already satisfied: numpy in /usr/local/lib/python3.7/dist-packages (from -r requirements.txt (line 2)) (1.19.5)\n",
            "Requirement already satisfied: Pillow in /usr/local/lib/python3.7/dist-packages (from -r requirements.txt (line 3)) (7.1.2)\n",
            "Requirement already satisfied: scipy in /usr/local/lib/python3.7/dist-packages (from -r requirements.txt (line 4)) (1.4.1)\n",
            "Requirement already satisfied: tensorboard in /usr/local/lib/python3.7/dist-packages (from -r requirements.txt (line 5)) (2.5.0)\n",
            "Collecting tensorboardx\n",
            "\u001b[?25l  Downloading https://files.pythonhosted.org/packages/99/0b/a26bbe92667c549d39c40b80c5ddec638fbae9521f04aeef26560e07e504/tensorboardX-2.4-py2.py3-none-any.whl (124kB)\n",
            "\u001b[K     |████████████████████████████████| 133kB 49.8MB/s \n",
            "\u001b[?25hRequirement already satisfied: tqdm in /usr/local/lib/python3.7/dist-packages (from -r requirements.txt (line 7)) (4.41.1)\n",
            "Collecting trimesh\n",
            "\u001b[?25l  Downloading https://files.pythonhosted.org/packages/30/0e/00743d0bbfad9b6aebf0206bd3f8d06386b8fc1b7203e30723687129bb12/trimesh-3.9.20-py3-none-any.whl (632kB)\n",
            "\u001b[K     |████████████████████████████████| 634kB 49.3MB/s \n",
            "\u001b[?25hRequirement already satisfied: astunparse in /usr/local/lib/python3.7/dist-packages (from jittor->-r requirements.txt (line 1)) (1.6.3)\n",
            "Requirement already satisfied: requests<3,>=2.21.0 in /usr/local/lib/python3.7/dist-packages (from tensorboard->-r requirements.txt (line 5)) (2.23.0)\n",
            "Requirement already satisfied: markdown>=2.6.8 in /usr/local/lib/python3.7/dist-packages (from tensorboard->-r requirements.txt (line 5)) (3.3.4)\n",
            "Requirement already satisfied: google-auth-oauthlib<0.5,>=0.4.1 in /usr/local/lib/python3.7/dist-packages (from tensorboard->-r requirements.txt (line 5)) (0.4.4)\n",
            "Requirement already satisfied: wheel>=0.26; python_version >= \"3\" in /usr/local/lib/python3.7/dist-packages (from tensorboard->-r requirements.txt (line 5)) (0.36.2)\n",
            "Requirement already satisfied: grpcio>=1.24.3 in /usr/local/lib/python3.7/dist-packages (from tensorboard->-r requirements.txt (line 5)) (1.34.1)\n",
            "Requirement already satisfied: tensorboard-data-server<0.7.0,>=0.6.0 in /usr/local/lib/python3.7/dist-packages (from tensorboard->-r requirements.txt (line 5)) (0.6.1)\n",
            "Requirement already satisfied: tensorboard-plugin-wit>=1.6.0 in /usr/local/lib/python3.7/dist-packages (from tensorboard->-r requirements.txt (line 5)) (1.8.0)\n",
            "Requirement already satisfied: google-auth<2,>=1.6.3 in /usr/local/lib/python3.7/dist-packages (from tensorboard->-r requirements.txt (line 5)) (1.31.0)\n",
            "Requirement already satisfied: protobuf>=3.6.0 in /usr/local/lib/python3.7/dist-packages (from tensorboard->-r requirements.txt (line 5)) (3.12.4)\n",
            "Requirement already satisfied: absl-py>=0.4 in /usr/local/lib/python3.7/dist-packages (from tensorboard->-r requirements.txt (line 5)) (0.12.0)\n",
            "Requirement already satisfied: setuptools>=41.0.0 in /usr/local/lib/python3.7/dist-packages (from tensorboard->-r requirements.txt (line 5)) (57.0.0)\n",
            "Requirement already satisfied: werkzeug>=0.11.15 in /usr/local/lib/python3.7/dist-packages (from tensorboard->-r requirements.txt (line 5)) (1.0.1)\n",
            "Requirement already satisfied: six<2.0,>=1.6.1 in /usr/local/lib/python3.7/dist-packages (from astunparse->jittor->-r requirements.txt (line 1)) (1.15.0)\n",
            "Requirement already satisfied: idna<3,>=2.5 in /usr/local/lib/python3.7/dist-packages (from requests<3,>=2.21.0->tensorboard->-r requirements.txt (line 5)) (2.10)\n",
            "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.7/dist-packages (from requests<3,>=2.21.0->tensorboard->-r requirements.txt (line 5)) (2021.5.30)\n",
            "Requirement already satisfied: urllib3!=1.25.0,!=1.25.1,<1.26,>=1.21.1 in /usr/local/lib/python3.7/dist-packages (from requests<3,>=2.21.0->tensorboard->-r requirements.txt (line 5)) (1.24.3)\n",
            "Requirement already satisfied: chardet<4,>=3.0.2 in /usr/local/lib/python3.7/dist-packages (from requests<3,>=2.21.0->tensorboard->-r requirements.txt (line 5)) (3.0.4)\n",
            "Requirement already satisfied: importlib-metadata; python_version < \"3.8\" in /usr/local/lib/python3.7/dist-packages (from markdown>=2.6.8->tensorboard->-r requirements.txt (line 5)) (4.5.0)\n",
            "Requirement already satisfied: requests-oauthlib>=0.7.0 in /usr/local/lib/python3.7/dist-packages (from google-auth-oauthlib<0.5,>=0.4.1->tensorboard->-r requirements.txt (line 5)) (1.3.0)\n",
            "Requirement already satisfied: rsa<5,>=3.1.4; python_version >= \"3.6\" in /usr/local/lib/python3.7/dist-packages (from google-auth<2,>=1.6.3->tensorboard->-r requirements.txt (line 5)) (4.7.2)\n",
            "Requirement already satisfied: cachetools<5.0,>=2.0.0 in /usr/local/lib/python3.7/dist-packages (from google-auth<2,>=1.6.3->tensorboard->-r requirements.txt (line 5)) (4.2.2)\n",
            "Requirement already satisfied: pyasn1-modules>=0.2.1 in /usr/local/lib/python3.7/dist-packages (from google-auth<2,>=1.6.3->tensorboard->-r requirements.txt (line 5)) (0.2.8)\n",
            "Requirement already satisfied: zipp>=0.5 in /usr/local/lib/python3.7/dist-packages (from importlib-metadata; python_version < \"3.8\"->markdown>=2.6.8->tensorboard->-r requirements.txt (line 5)) (3.4.1)\n",
            "Requirement already satisfied: typing-extensions>=3.6.4; python_version < \"3.8\" in /usr/local/lib/python3.7/dist-packages (from importlib-metadata; python_version < \"3.8\"->markdown>=2.6.8->tensorboard->-r requirements.txt (line 5)) (3.7.4.3)\n",
            "Requirement already satisfied: oauthlib>=3.0.0 in /usr/local/lib/python3.7/dist-packages (from requests-oauthlib>=0.7.0->google-auth-oauthlib<0.5,>=0.4.1->tensorboard->-r requirements.txt (line 5)) (3.1.1)\n",
            "Requirement already satisfied: pyasn1>=0.1.3 in /usr/local/lib/python3.7/dist-packages (from rsa<5,>=3.1.4; python_version >= \"3.6\"->google-auth<2,>=1.6.3->tensorboard->-r requirements.txt (line 5)) (0.4.8)\n",
            "Building wheels for collected packages: jittor\n",
            "  Building wheel for jittor (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
            "  Created wheel for jittor: filename=jittor-1.2.3.55-cp37-none-any.whl size=1693684 sha256=e2b6d1ba71e88d4875314f8e29020ebc0e1eaf07558cdcf5f0d2086818805bf2\n",
            "  Stored in directory: /root/.cache/pip/wheels/72/25/39/00c71695ff247a1dc5101802794c63a9a1e5fec01cb38e8c77\n",
            "Successfully built jittor\n",
            "Installing collected packages: jittor, tensorboardx, trimesh\n",
            "Successfully installed jittor-1.2.3.55 tensorboardx-2.4 trimesh-3.9.20\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "tLEOvGSgKU_n",
        "outputId": "c44b4b22-45e1-40db-ec82-2dbcd902ab9b"
      },
      "source": [
        "!sh scripts/shrec11-split10/get_data.sh\n",
        "!sh scripts/shrec11-split16/get_data.sh\n",
        "!sh scripts/cubes/get_data.sh\n",
        "!sh scripts/manifold40/get_data.sh\n",
        "!sh scripts/humanbody/get_data.sh\n",
        "!sh scripts/coseg-aliens/get_data.sh"
      ],
      "execution_count": 12,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "--2021-07-03 21:00:22--  https://cloud.tsinghua.edu.cn/f/6e59181fd26d4a2dbad9/?dl=1\n",
            "Resolving cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)... 101.6.8.7\n",
            "Connecting to cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)|101.6.8.7|:443... connected.\n",
            "HTTP request sent, awaiting response... 302 Found\n",
            "Location: https://cloud.tsinghua.edu.cn/seafhttp/files/331a096a-93a8-49fa-8021-c9fec36090a2/SHREC11-MAPS-48-4-split10.zip [following]\n",
            "--2021-07-03 21:00:24--  https://cloud.tsinghua.edu.cn/seafhttp/files/331a096a-93a8-49fa-8021-c9fec36090a2/SHREC11-MAPS-48-4-split10.zip\n",
            "Reusing existing connection to cloud.tsinghua.edu.cn:443.\n",
            "HTTP request sent, awaiting response... 200 OK\n",
            "Length: 2172924462 (2.0G) [application/zip]\n",
            "Saving to: ‘SHREC11-MAPS-48-4-split10.zip’\n",
            "\n",
            "SHREC11-MAPS-48-4-s 100%[===================>]   2.02G  8.21MB/s    in 4m 9s   \n",
            "\n",
            "2021-07-03 21:04:33 (8.31 MB/s) - ‘SHREC11-MAPS-48-4-split10.zip’ saved [2172924462/2172924462]\n",
            "\n",
            "downloaded the data and putting it in:  scripts/shrec11-split10/../../data\n",
            "unzipping\n",
            "--2021-07-03 21:07:15--  https://cloud.tsinghua.edu.cn/f/e8f71bf910af4b1798c1/?dl=1\n",
            "Resolving cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)... 101.6.8.7\n",
            "Connecting to cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)|101.6.8.7|:443... connected.\n",
            "HTTP request sent, awaiting response... 302 Found\n",
            "Location: https://cloud.tsinghua.edu.cn/seafhttp/files/829b973f-240c-4b56-8105-df9bb8e0d532/SHREC11-MAPS-48-4-split16.zip [following]\n",
            "--2021-07-03 21:07:16--  https://cloud.tsinghua.edu.cn/seafhttp/files/829b973f-240c-4b56-8105-df9bb8e0d532/SHREC11-MAPS-48-4-split16.zip\n",
            "Reusing existing connection to cloud.tsinghua.edu.cn:443.\n",
            "HTTP request sent, awaiting response... 200 OK\n",
            "Length: 2172932022 (2.0G) [application/zip]\n",
            "Saving to: ‘SHREC11-MAPS-48-4-split16.zip’\n",
            "\n",
            "SHREC11-MAPS-48-4-s 100%[===================>]   2.02G  8.10MB/s    in 4m 23s  \n",
            "\n",
            "2021-07-03 21:11:40 (7.87 MB/s) - ‘SHREC11-MAPS-48-4-split16.zip’ saved [2172932022/2172932022]\n",
            "\n",
            "downloaded the data and putting it in:  scripts/shrec11-split16/../../data\n",
            "unzipping\n",
            "--2021-07-03 21:14:23--  https://cloud.tsinghua.edu.cn/f/f1d091376e9e4bd69ce1/?dl=1\n",
            "Resolving cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)... 101.6.8.7\n",
            "Connecting to cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)|101.6.8.7|:443... connected.\n",
            "HTTP request sent, awaiting response... 302 Found\n",
            "Location: https://cloud.tsinghua.edu.cn/seafhttp/files/18b598f6-a04a-4cb9-a875-428f99f0a40f/Cubes-MAPS-48-4.zip [following]\n",
            "--2021-07-03 21:14:24--  https://cloud.tsinghua.edu.cn/seafhttp/files/18b598f6-a04a-4cb9-a875-428f99f0a40f/Cubes-MAPS-48-4.zip\n",
            "Reusing existing connection to cloud.tsinghua.edu.cn:443.\n",
            "HTTP request sent, awaiting response... 200 OK\n",
            "Length: 7067788386 (6.6G) [application/zip]\n",
            "Saving to: ‘Cubes-MAPS-48-4.zip’\n",
            "\n",
            "Cubes-MAPS-48-4.zip 100%[===================>]   6.58G  8.21MB/s    in 14m 12s \n",
            "\n",
            "2021-07-03 21:28:37 (7.91 MB/s) - ‘Cubes-MAPS-48-4.zip’ saved [7067788386/7067788386]\n",
            "\n",
            "downloaded the data and putting it in:  scripts/cubes/../../data\n",
            "unzipping\n",
            "--2021-07-03 21:39:08--  https://cloud.tsinghua.edu.cn/f/af5c682587cc4f9da9b8/?dl=1\n",
            "Resolving cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)... 101.6.8.7\n",
            "Connecting to cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)|101.6.8.7|:443... connected.\n",
            "HTTP request sent, awaiting response... 302 Found\n",
            "Location: https://cloud.tsinghua.edu.cn/seafhttp/files/1017f34d-8c5e-4929-b0e5-bbf03d4cfad5/Manifold40-MAPS-96-3.zip [following]\n",
            "--2021-07-03 21:39:10--  https://cloud.tsinghua.edu.cn/seafhttp/files/1017f34d-8c5e-4929-b0e5-bbf03d4cfad5/Manifold40-MAPS-96-3.zip\n",
            "Reusing existing connection to cloud.tsinghua.edu.cn:443.\n",
            "HTTP request sent, awaiting response... 200 OK\n",
            "Length: 10761967289 (10G) [application/zip]\n",
            "Saving to: ‘Manifold40-MAPS-96-3.zip’\n",
            "\n",
            "Manifold40-MAPS-96-   2%[                    ] 254.73M  2.25MB/s    in 44s     \n",
            "\n",
            "2021-07-03 21:39:54 (5.79 MB/s) - Connection closed at byte 267108352. Retrying.\n",
            "\n",
            "--2021-07-03 21:39:55--  (try: 2)  https://cloud.tsinghua.edu.cn/seafhttp/files/1017f34d-8c5e-4929-b0e5-bbf03d4cfad5/Manifold40-MAPS-96-3.zip\n",
            "Connecting to cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)|101.6.8.7|:443... connected.\n",
            "HTTP request sent, awaiting response... 206 Partial Content\n",
            "Length: 10761967289 (10G), 10494858937 (9.8G) remaining [application/zip]\n",
            "Saving to: ‘Manifold40-MAPS-96-3.zip’\n",
            "\n",
            "Manifold40-MAPS-96- 100%[===================>]  10.02G  8.17MB/s    in 21m 10s \n",
            "\n",
            "2021-07-03 22:01:06 (7.88 MB/s) - ‘Manifold40-MAPS-96-3.zip’ saved [10761967289/10761967289]\n",
            "\n",
            "downloaded the data and putting it in:  scripts/manifold40/../../data\n",
            "unzipping\n",
            "--2021-07-03 23:07:00--  https://cloud.tsinghua.edu.cn/f/305c270079724f689418/?dl=1\n",
            "Resolving cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)... 101.6.8.7\n",
            "Connecting to cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)|101.6.8.7|:443... connected.\n",
            "HTTP request sent, awaiting response... 302 Found\n",
            "Location: https://cloud.tsinghua.edu.cn/seafhttp/files/b7f7148f-a5da-402d-b100-31acf4874889/HumanBody-NS-256-3.zip [following]\n",
            "--2021-07-03 23:07:02--  https://cloud.tsinghua.edu.cn/seafhttp/files/b7f7148f-a5da-402d-b100-31acf4874889/HumanBody-NS-256-3.zip\n",
            "Reusing existing connection to cloud.tsinghua.edu.cn:443.\n",
            "HTTP request sent, awaiting response... 200 OK\n",
            "Length: 1642287848 (1.5G) [application/zip]\n",
            "Saving to: ‘HumanBody-NS-256-3.zip’\n",
            "\n",
            "HumanBody-NS-256-3. 100%[===================>]   1.53G  6.51MB/s    in 3m 25s  \n",
            "\n",
            "2021-07-03 23:10:27 (7.62 MB/s) - ‘HumanBody-NS-256-3.zip’ saved [1642287848/1642287848]\n",
            "\n",
            "downloaded the data and putting it in:  scripts/humanbody/../../data\n",
            "unzipping\n",
            "--2021-07-03 23:18:49--  https://cloud.tsinghua.edu.cn/f/27cc19807ff644989881/?dl=1\n",
            "Resolving cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)... 101.6.8.7\n",
            "Connecting to cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)|101.6.8.7|:443... connected.\n",
            "HTTP request sent, awaiting response... 302 Found\n",
            "Location: https://cloud.tsinghua.edu.cn/seafhttp/files/462ccc43-8f6d-4307-8e4f-cfb4b33a5ee3/coseg-aliens-MAPS-256-3.zip [following]\n",
            "--2021-07-03 23:18:51--  https://cloud.tsinghua.edu.cn/seafhttp/files/462ccc43-8f6d-4307-8e4f-cfb4b33a5ee3/coseg-aliens-MAPS-256-3.zip\n",
            "Reusing existing connection to cloud.tsinghua.edu.cn:443.\n",
            "HTTP request sent, awaiting response... 200 OK\n",
            "Length: 573317535 (547M) [application/zip]\n",
            "Saving to: ‘coseg-aliens-MAPS-256-3.zip’\n",
            "\n",
            "coseg-aliens-MAPS-2 100%[===================>] 546.76M  6.79MB/s    in 75s     \n",
            "\n",
            "2021-07-03 23:20:06 (7.32 MB/s) - ‘coseg-aliens-MAPS-256-3.zip’ saved [573317535/573317535]\n",
            "\n",
            "downloaded the data and putting it in:  scripts/coseg-aliens/../../data\n",
            "unzipping\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "ulZE2JhSKgIG",
        "outputId": "803c2c21-7d73-4a2f-9f03-4f66930b6d9a"
      },
      "source": [
        "!sh scripts/shrec11-split10/get_pretrained.sh\n",
        "!sh scripts/shrec11-split16/get_pretrained.sh\n",
        "!sh scripts/cubes/get_pretrained.sh\n",
        "!sh scripts/manifold40/get_pretrained.sh\n",
        "!sh scripts/humanbody/get_pretrained.sh\n",
        "!sh scripts/coseg-aliens/get_pretrained.sh"
      ],
      "execution_count": 13,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "--2021-07-03 23:26:08--  https://cloud.tsinghua.edu.cn/f/b1ac19339dd14c4d9631/?dl=1\n",
            "Resolving cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)... 101.6.8.7\n",
            "Connecting to cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)|101.6.8.7|:443... connected.\n",
            "HTTP request sent, awaiting response... 302 Found\n",
            "Location: https://cloud.tsinghua.edu.cn/seafhttp/files/e605cd5f-68fa-4e0c-a287-0b4537266140/shrec11-split10.pkl [following]\n",
            "--2021-07-03 23:26:09--  https://cloud.tsinghua.edu.cn/seafhttp/files/e605cd5f-68fa-4e0c-a287-0b4537266140/shrec11-split10.pkl\n",
            "Reusing existing connection to cloud.tsinghua.edu.cn:443.\n",
            "HTTP request sent, awaiting response... 200 OK\n",
            "Length: 1156336 (1.1M) [application/octet-stream]\n",
            "Saving to: ‘shrec11-split10.pkl’\n",
            "\n",
            "shrec11-split10.pkl 100%[===================>]   1.10M   594KB/s    in 1.9s    \n",
            "\n",
            "2021-07-03 23:26:11 (594 KB/s) - ‘shrec11-split10.pkl’ saved [1156336/1156336]\n",
            "\n",
            "downloaded the checkpoint and putting it in:  scripts/shrec11-split10/../../checkpoints\n",
            "--2021-07-03 23:26:11--  https://cloud.tsinghua.edu.cn/f/5cbb6ddc835f48d9a6be/?dl=1\n",
            "Resolving cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)... 101.6.8.7\n",
            "Connecting to cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)|101.6.8.7|:443... connected.\n",
            "HTTP request sent, awaiting response... 302 Found\n",
            "Location: https://cloud.tsinghua.edu.cn/seafhttp/files/43931a93-1601-4a13-a49f-e185a8b27d9c/shrec11-split16.pkl [following]\n",
            "--2021-07-03 23:26:13--  https://cloud.tsinghua.edu.cn/seafhttp/files/43931a93-1601-4a13-a49f-e185a8b27d9c/shrec11-split16.pkl\n",
            "Reusing existing connection to cloud.tsinghua.edu.cn:443.\n",
            "HTTP request sent, awaiting response... 200 OK\n",
            "Length: 1903865 (1.8M) [application/octet-stream]\n",
            "Saving to: ‘shrec11-split16.pkl’\n",
            "\n",
            "shrec11-split16.pkl 100%[===================>]   1.82M   867KB/s    in 2.1s    \n",
            "\n",
            "2021-07-03 23:26:16 (867 KB/s) - ‘shrec11-split16.pkl’ saved [1903865/1903865]\n",
            "\n",
            "downloaded the checkpoint and putting it in:  scripts/shrec11-split16/../../checkpoints\n",
            "--2021-07-03 23:26:16--  https://cloud.tsinghua.edu.cn/f/b7b733157e7d44ea9bca/?dl=1\n",
            "Resolving cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)... 101.6.8.7\n",
            "Connecting to cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)|101.6.8.7|:443... connected.\n",
            "HTTP request sent, awaiting response... 302 Found\n",
            "Location: https://cloud.tsinghua.edu.cn/seafhttp/files/47795067-dd41-4c4b-9e21-125d29af623c/Cubes.pkl [following]\n",
            "--2021-07-03 23:26:17--  https://cloud.tsinghua.edu.cn/seafhttp/files/47795067-dd41-4c4b-9e21-125d29af623c/Cubes.pkl\n",
            "Reusing existing connection to cloud.tsinghua.edu.cn:443.\n",
            "HTTP request sent, awaiting response... 200 OK\n",
            "Length: 1899737 (1.8M) [application/octet-stream]\n",
            "Saving to: ‘Cubes.pkl’\n",
            "\n",
            "Cubes.pkl           100%[===================>]   1.81M   751KB/s    in 2.5s    \n",
            "\n",
            "2021-07-03 23:26:20 (751 KB/s) - ‘Cubes.pkl’ saved [1899737/1899737]\n",
            "\n",
            "downloaded the checkpoint and putting it in:  scripts/cubes/../../checkpoints\n",
            "--2021-07-03 23:26:20--  https://cloud.tsinghua.edu.cn/f/6f6d9137bb2d475e98c5/?dl=1\n",
            "Resolving cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)... 101.6.8.7\n",
            "Connecting to cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)|101.6.8.7|:443... connected.\n",
            "HTTP request sent, awaiting response... 302 Found\n",
            "Location: https://cloud.tsinghua.edu.cn/seafhttp/files/83750a41-c0ca-47fc-a44d-260b8371a0fb/Manifold40.pkl [following]\n",
            "--2021-07-03 23:26:22--  https://cloud.tsinghua.edu.cn/seafhttp/files/83750a41-c0ca-47fc-a44d-260b8371a0fb/Manifold40.pkl\n",
            "Reusing existing connection to cloud.tsinghua.edu.cn:443.\n",
            "HTTP request sent, awaiting response... 200 OK\n",
            "Length: 3439033 (3.3M) [application/octet-stream]\n",
            "Saving to: ‘Manifold40.pkl’\n",
            "\n",
            "Manifold40.pkl      100%[===================>]   3.28M  1.41MB/s    in 2.3s    \n",
            "\n",
            "2021-07-03 23:26:24 (1.41 MB/s) - ‘Manifold40.pkl’ saved [3439033/3439033]\n",
            "\n",
            "downloaded the checkpoint and putting it in:  scripts/manifold40/../../checkpoints\n",
            "--2021-07-03 23:26:24--  https://cloud.tsinghua.edu.cn/f/1f86eee93a1e4dd7a690/?dl=1\n",
            "Resolving cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)... 101.6.8.7\n",
            "Connecting to cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)|101.6.8.7|:443... connected.\n",
            "HTTP request sent, awaiting response... 302 Found\n",
            "Location: https://cloud.tsinghua.edu.cn/seafhttp/files/403c0ee6-508e-4364-b660-d1f065c8c4d4/HumanBody.pkl [following]\n",
            "--2021-07-03 23:26:26--  https://cloud.tsinghua.edu.cn/seafhttp/files/403c0ee6-508e-4364-b660-d1f065c8c4d4/HumanBody.pkl\n",
            "Reusing existing connection to cloud.tsinghua.edu.cn:443.\n",
            "HTTP request sent, awaiting response... 200 OK\n",
            "Length: 43027193 (41M) [application/octet-stream]\n",
            "Saving to: ‘HumanBody.pkl’\n",
            "\n",
            "HumanBody.pkl       100%[===================>]  41.03M  8.76MB/s    in 6.8s    \n",
            "\n",
            "2021-07-03 23:26:34 (6.00 MB/s) - ‘HumanBody.pkl’ saved [43027193/43027193]\n",
            "\n",
            "downloaded the checkpoint and putting it in:  scripts/humanbody/../../checkpoints\n",
            "--2021-07-03 23:26:34--  https://cloud.tsinghua.edu.cn/f/60ef71b57a514e4a8dbb/?dl=1\n",
            "Resolving cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)... 101.6.8.7\n",
            "Connecting to cloud.tsinghua.edu.cn (cloud.tsinghua.edu.cn)|101.6.8.7|:443... connected.\n",
            "HTTP request sent, awaiting response... 302 Found\n",
            "Location: https://cloud.tsinghua.edu.cn/seafhttp/files/09b30b79-d9f5-4e2b-9818-6cd884036c50/coseg-aliens.pkl [following]\n",
            "--2021-07-03 23:26:35--  https://cloud.tsinghua.edu.cn/seafhttp/files/09b30b79-d9f5-4e2b-9818-6cd884036c50/coseg-aliens.pkl\n",
            "Reusing existing connection to cloud.tsinghua.edu.cn:443.\n",
            "HTTP request sent, awaiting response... 200 OK\n",
            "Length: 43025129 (41M) [application/octet-stream]\n",
            "Saving to: ‘coseg-aliens.pkl’\n",
            "\n",
            "coseg-aliens.pkl    100%[===================>]  41.03M  8.15MB/s    in 6.9s    \n",
            "\n",
            "2021-07-03 23:26:43 (5.92 MB/s) - ‘coseg-aliens.pkl’ saved [43025129/43025129]\n",
            "\n",
            "downloaded the checkpoint and putting it in:  scripts/coseg-aliens/../../checkpoints\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "m6w1Czn2sgZK",
        "outputId": "5895f0e3-45ad-4b97-bf4b-6b2a874dbec2"
      },
      "source": [
        "!sh scripts/shrec11-split10/test.sh\n",
        "!sh scripts/shrec11-split16/test.sh\n",
        "!sh scripts/cubes/test.sh\n",
        "!sh scripts/manifold40/test.sh\n",
        "!sh scripts/humanbody/test.sh\n",
        "!sh scripts/coseg-aliens/test.sh"
      ],
      "execution_count": 14,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "[i 0703 23:28:51.061375 20 lock.py:58] Create lock file:/root/.cache/jittor/default/jittor.lock\n",
            "[i 0703 23:28:51.073818 20 compiler.py:869] Jittor(1.2.3.55) src: /usr/local/lib/python3.7/dist-packages/jittor\n",
            "[i 0703 23:28:51.087680 20 compiler.py:870] g++ at /usr/bin/g++(7.5.0)\n",
            "[i 0703 23:28:51.087860 20 compiler.py:871] cache_path: /root/.cache/jittor/default/g++\n",
            "[i 0703 23:28:51.106178 20 __init__.py:286] Found nvcc(11.0.221) at /usr/local/cuda/bin/nvcc.\n",
            "[i 0703 23:28:51.143630 20 __init__.py:286] Found addr2line(2.30) at /usr/bin/addr2line.\n",
            "[i 0703 23:28:51.186981 20 compiler.py:958] py_include: -I/usr/include/python3.7m -I/usr/include/python3.7m\n",
            "[i 0703 23:28:51.221350 20 compiler.py:960] extension_suffix: .cpython-37m-x86_64-linux-gnu.so\n",
            "[i 0703 23:28:51.221552 20 compiler.py:32] Create cache dir: /root/.cache/jittor/default/g++/jit\n",
            "[i 0703 23:28:51.221736 20 compiler.py:32] Create cache dir: /root/.cache/jittor/default/g++/obj_files\n",
            "[i 0703 23:28:51.221881 20 compiler.py:32] Create cache dir: /root/.cache/jittor/default/g++/gen\n",
            "[i 0703 23:28:51.222020 20 compiler.py:32] Create cache dir: /root/.cache/jittor/default/g++/checkpoints\n",
            "\u001b[38;5;2m[i 0703 23:29:03.827900 60 compiler.py:1093] OS type:ubuntu OS key:ubuntu\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:29:07.244357 60 __init__.py:178] Total mem: 25.46GB, using 8 procs for compiling.\u001b[m\n",
            "Compiling jittor_core(144/144) used: 80.661s eta: 0.000s\n",
            "\u001b[38;5;2m[i 0703 23:30:28.440691 60 jit_compiler.cc:22] Load cc_path: /usr/bin/g++\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:30:28.441914 60 init.cc:55] Found cuda archs: [70,]\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:30:28.523907 60 __init__.py:286] Found mpicc(2.1.1) at /usr/bin/mpicc.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:30:28.584573 60 compiler.py:32] Create cache dir: /root/.cache/jittor/default/g++/custom_ops\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:30:28.586526 60 compiler.py:667] handle pyjt_include/usr/local/lib/python3.7/dist-packages/jittor/extern/mpi/inc/mpi_warper.h\u001b[m\n",
            "Compiling jittor_mpi_core(7/7) used: 3.471s eta: 0.000s\n",
            "\u001b[38;5;2m[i 0703 23:30:32.148102 60 compiler.py:32] Create cache dir: /root/.cache/jittor/nccl\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:30:32.157137 60 compile_extern.py:347] Downloading nccl...\u001b[m\n",
            "Downloading https://codeload.github.com/NVIDIA/nccl/tar.gz/v2.8.4-1 to /root/.cache/jittor/nccl/nccl.tgz\n",
            "204800it [00:00, 891333.72it/s]\n",
            "\u001b[38;5;2m[i 0703 23:30:32.389121 60 compiler.py:32] Create cache dir: /root/.cache/jittor/cutt\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:30:32.389369 60 compile_extern.py:270] Downloading cutt...\u001b[m\n",
            "Downloading https://codeload.github.com/Jittor/cutt/zip/v1.1 to /root/.cache/jittor/cutt/cutt-1.1.zip\n",
            "303104it [00:00, 1351110.99it/s]\n",
            "\u001b[38;5;2m[i 0703 23:30:32.627806 60 compile_extern.py:283] installing cutt...\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:31:01.045440 60 compiler.py:32] Create cache dir: /root/.cache/jittor/mkl\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:31:01.045670 60 compile_extern.py:47] Downloading mkl...\u001b[m\n",
            "Downloading https://cg.cs.tsinghua.edu.cn/jittor/assets/dnnl_lnx_2.2.0_cpu_gomp.tgz to /root/.cache/jittor/mkl/dnnl_lnx_2.2.0_cpu_gomp.tgz\n",
            " 99% 10969088/11053466 [00:20<00:00, 657745.33it/s]Use time: 26.86 ms per iteration.\n",
            "Example passed on CPU.\n",
            "Compiling gen_ops_mkl_matmul_mkl_conv_backward_x_mkl_conv_mkl_conv_backward_w_mkl_test(7/7) used: 5.211s eta: 0.000s\n",
            "\u001b[38;5;2m[i 0703 23:31:33.073776 60 compiler.py:32] Create cache dir: /root/.cache/jittor/default/g++/cuda\u001b[m\n",
            "Compiling cuda_extern(3/3) used: 2.028s eta: 0.000s\n",
            "Compiling gen_ops_cub_where_cub_argsort_cub_arg_reduce_cub_test(5/5) used: 2.844s eta: 0.000s\n",
            "\u001b[38;5;2m[i 0703 23:31:38.123181 60 compile_extern.py:20] found /usr/local/cuda/include/cublas.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:31:38.138169 60 compile_extern.py:20] found /usr/local/cuda/lib64/libcublas.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:31:38.138331 60 compile_extern.py:20] found /usr/local/cuda/lib64/libcublasLt.so.11\u001b[m\n",
            "11059200it [00:40, 657745.33it/s]                  Compiling gen_ops_cublas_test_cublas_matmul_cublas_batched_matmul(7/7) used: 3.010s eta: 0.000s\n",
            "\u001b[38;5;2m[i 0703 23:31:42.723974 60 compile_extern.py:20] found /usr/include/cudnn.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:31:42.747441 60 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:31:42.748339 60 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_ops_infer.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:31:42.789337 60 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_ops_train.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:31:42.801167 60 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_cnn_infer.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:31:53.519261 60 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_cnn_train.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:31:53.653522 60 compiler.py:667] handle pyjt_include/usr/local/lib/python3.7/dist-packages/jittor/extern/cuda/cudnn/inc/cudnn_warper.h\u001b[m\n",
            "Compiling gen_ops_cudnn_conv3d_cudnn_conv_backward_x_cudnn_test_cudnn_conv3d_backward_x_cu___hash7b2053b7af34de53e3e71975a0b47617(12/12) used: 6.231s eta: 0.000s\n",
            "\u001b[38;5;2m[i 0703 23:32:03.084470 60 compile_extern.py:20] found /usr/local/cuda/include/curand.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:32:03.179742 60 compile_extern.py:20] found /usr/local/cuda/lib64/libcurand.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:32:06.830776 60 cuda_flags.cc:26] CUDA enabled.\u001b[m\n",
            "name:  shrec11-split10\n",
            "\n",
            "Compiling Operators(7/7) used: 11.4s eta:    0s \n",
            "\n",
            "Test 0:   0% 0/99 [00:00<?, ?it/s]\u001b[A\n",
            "Compiling Operators(1/1) used:  5.3s eta:    0s \n",
            "\n",
            "Compiling Operators(1/1) used:    4s eta:    0s \n",
            "\n",
            "Compiling Operators(1/1) used: 5.38s eta:    0s \n",
            "\n",
            "Compiling Operators(1/1) used: 5.34s eta:    0s \n",
            "\n",
            "Compiling Operators(37/37) used: 66.4s eta:    0s \n",
            "\n",
            "Test 0:   1% 1/99 [01:34<2:35:07, 94.98s/it]\u001b[A\n",
            "Test 0:   2% 2/99 [01:35<1:47:49, 66.70s/it]\u001b[A\n",
            "Test 0:   3% 3/99 [01:36<1:14:58, 46.86s/it]\u001b[A\n",
            "Test 0:   4% 4/99 [01:36<52:13, 32.98s/it]  \u001b[A\n",
            "Test 0:   5% 5/99 [01:37<36:27, 23.27s/it]\u001b[A\n",
            "Test 0:   6% 6/99 [01:38<25:31, 16.46s/it]\u001b[A\n",
            "Test 0:   7% 7/99 [01:38<17:57, 11.71s/it]\u001b[A\n",
            "Test 0:   8% 8/99 [01:39<12:44,  8.40s/it]\u001b[A\n",
            "Test 0:   9% 9/99 [01:40<09:07,  6.08s/it]\u001b[A\n",
            "Test 0:  10% 10/99 [01:40<06:37,  4.46s/it]\u001b[A\n",
            "Test 0:  11% 11/99 [01:41<04:50,  3.31s/it]\u001b[A\n",
            "Test 0:  12% 12/99 [01:41<03:37,  2.50s/it]\u001b[A\n",
            "Test 0:  13% 13/99 [01:42<02:47,  1.95s/it]\u001b[A\n",
            "Test 0:  14% 14/99 [01:43<02:09,  1.53s/it]\u001b[A\n",
            "Test 0:  15% 15/99 [01:43<01:45,  1.25s/it]\u001b[A\n",
            "Test 0:  16% 16/99 [01:44<01:29,  1.08s/it]\u001b[A\n",
            "Test 0:  17% 17/99 [01:45<01:17,  1.05it/s]\u001b[A\n",
            "Test 0:  18% 18/99 [01:45<01:09,  1.16it/s]\u001b[A\n",
            "Test 0:  19% 19/99 [01:46<01:03,  1.25it/s]\u001b[A\n",
            "Test 0:  20% 20/99 [01:47<01:01,  1.29it/s]\u001b[A\n",
            "Test 0:  21% 21/99 [01:47<00:58,  1.33it/s]\u001b[A\n",
            "Test 0:  22% 22/99 [01:48<00:54,  1.43it/s]\u001b[A\n",
            "Test 0:  23% 23/99 [01:49<00:53,  1.42it/s]\u001b[A\n",
            "Test 0:  24% 24/99 [01:49<00:52,  1.43it/s]\u001b[A\n",
            "Test 0:  25% 25/99 [01:50<00:52,  1.42it/s]\u001b[A\n",
            "Test 0:  26% 26/99 [01:52<01:15,  1.03s/it]\u001b[A\n",
            "Test 0:  27% 27/99 [01:53<01:13,  1.02s/it]\u001b[A\n",
            "Test 0:  28% 28/99 [01:54<01:26,  1.22s/it]\u001b[A\n",
            "Test 0:  29% 29/99 [01:57<01:50,  1.58s/it]\u001b[A\n",
            "Test 0:  30% 30/99 [01:59<02:03,  1.79s/it]\u001b[A\n",
            "Test 0:  31% 31/99 [02:00<01:49,  1.61s/it]\u001b[A\n",
            "Test 0:  32% 32/99 [02:02<01:39,  1.49s/it]\u001b[A\n",
            "Test 0:  33% 33/99 [02:04<02:00,  1.83s/it]\u001b[A\n",
            "Test 0:  34% 34/99 [02:06<02:03,  1.89s/it]\u001b[A\n",
            "Test 0:  35% 35/99 [02:08<01:50,  1.73s/it]\u001b[A\n",
            "Test 0:  36% 36/99 [02:09<01:42,  1.63s/it]\u001b[A\n",
            "Test 0:  37% 37/99 [02:11<01:55,  1.87s/it]\u001b[A\n",
            "Test 0:  38% 38/99 [02:14<01:59,  1.95s/it]\u001b[A\n",
            "Test 0:  39% 39/99 [02:15<01:47,  1.80s/it]\u001b[A\n",
            "Test 0:  40% 40/99 [02:16<01:38,  1.67s/it]\u001b[A\n",
            "Test 0:  41% 41/99 [02:19<01:51,  1.91s/it]\u001b[A\n",
            "Test 0:  42% 42/99 [02:21<01:52,  1.97s/it]\u001b[A\n",
            "Test 0:  43% 43/99 [02:22<01:39,  1.78s/it]\u001b[A\n",
            "Test 0:  44% 44/99 [02:23<01:25,  1.55s/it]\u001b[A\n",
            "Test 0:  45% 45/99 [02:26<01:44,  1.93s/it]\u001b[A\n",
            "Test 0:  46% 46/99 [02:28<01:46,  2.01s/it]\u001b[A\n",
            "Test 0:  47% 47/99 [02:30<01:33,  1.80s/it]\u001b[A\n",
            "Test 0:  48% 48/99 [02:31<01:21,  1.60s/it]\u001b[A\n",
            "Test 0:  49% 49/99 [02:34<01:39,  1.98s/it]\u001b[A\n",
            "Test 0:  51% 50/99 [02:36<01:36,  1.96s/it]\u001b[A\n",
            "Test 0:  52% 51/99 [02:37<01:22,  1.73s/it]\u001b[A\n",
            "Test 0:  53% 52/99 [02:38<01:17,  1.65s/it]\u001b[A\n",
            "Test 0:  54% 53/99 [02:41<01:33,  2.04s/it]\u001b[A\n",
            "Test 0:  55% 54/99 [02:43<01:29,  1.99s/it]\u001b[A\n",
            "Test 0:  56% 55/99 [02:44<01:20,  1.83s/it]\u001b[A\n",
            "Test 0:  57% 56/99 [02:46<01:10,  1.65s/it]\u001b[A\n",
            "Test 0:  58% 57/99 [02:48<01:23,  1.98s/it]\u001b[A\n",
            "Test 0:  59% 58/99 [02:50<01:19,  1.94s/it]\u001b[A\n",
            "Test 0:  60% 59/99 [02:52<01:14,  1.87s/it]\u001b[A\n",
            "Test 0:  61% 60/99 [02:53<01:03,  1.63s/it]\u001b[A\n",
            "Test 0:  62% 61/99 [02:56<01:16,  2.02s/it]\u001b[A\n",
            "Test 0:  63% 62/99 [02:58<01:11,  1.93s/it]\u001b[A\n",
            "Test 0:  64% 63/99 [02:59<01:03,  1.77s/it]\u001b[A\n",
            "Test 0:  65% 64/99 [03:01<00:57,  1.65s/it]\u001b[A\n",
            "Test 0:  66% 65/99 [03:03<01:05,  1.94s/it]\u001b[A\n",
            "Test 0:  67% 66/99 [03:05<01:03,  1.93s/it]\u001b[A\n",
            "Test 0:  68% 67/99 [03:06<00:55,  1.74s/it]\u001b[A\n",
            "Test 0:  69% 68/99 [03:08<00:50,  1.62s/it]\u001b[A\n",
            "Test 0:  70% 69/99 [03:10<00:56,  1.89s/it]\u001b[A\n",
            "Test 0:  71% 70/99 [03:12<00:56,  1.95s/it]\u001b[A\n",
            "Test 0:  72% 71/99 [03:13<00:46,  1.67s/it]\u001b[A\n",
            "Test 0:  73% 72/99 [03:14<00:41,  1.52s/it]\u001b[A\n",
            "Test 0:  74% 73/99 [03:18<00:52,  2.01s/it]\u001b[A\n",
            "Test 0:  75% 74/99 [03:19<00:48,  1.94s/it]\u001b[A\n",
            "Test 0:  76% 75/99 [03:21<00:40,  1.70s/it]\u001b[A\n",
            "Test 0:  77% 76/99 [03:22<00:34,  1.52s/it]\u001b[A\n",
            "Test 0:  78% 77/99 [03:25<00:42,  1.95s/it]\u001b[A\n",
            "Test 0:  79% 78/99 [03:27<00:41,  1.96s/it]\u001b[A\n",
            "Test 0:  80% 79/99 [03:28<00:34,  1.72s/it]\u001b[A\n",
            "Test 0:  81% 80/99 [03:29<00:30,  1.63s/it]\u001b[A\n",
            "Test 0:  82% 81/99 [03:32<00:34,  1.92s/it]\u001b[A\n",
            "Test 0:  83% 82/99 [03:34<00:32,  1.91s/it]\u001b[A\n",
            "Test 0:  84% 83/99 [03:35<00:25,  1.62s/it]\u001b[A\n",
            "Test 0:  85% 84/99 [03:36<00:23,  1.59s/it]\u001b[A\n",
            "Test 0:  86% 85/99 [03:39<00:26,  1.93s/it]\u001b[A\n",
            "Test 0:  87% 86/99 [03:41<00:24,  1.90s/it]\u001b[A\n",
            "Test 0:  88% 87/99 [03:42<00:19,  1.66s/it]\u001b[A\n",
            "Test 0:  89% 88/99 [03:43<00:18,  1.64s/it]\u001b[A\n",
            "Test 0:  90% 89/99 [03:46<00:19,  1.98s/it]\u001b[A\n",
            "Test 0:  91% 90/99 [03:48<00:17,  1.90s/it]\u001b[A\n",
            "Test 0:  92% 91/99 [03:49<00:12,  1.59s/it]\u001b[A\n",
            "Test 0:  93% 92/99 [03:50<00:11,  1.62s/it]\u001b[A\n",
            "Test 0:  94% 93/99 [03:53<00:11,  1.86s/it]\u001b[A\n",
            "Test 0:  95% 94/99 [03:55<00:09,  1.81s/it]\u001b[A\n",
            "Test 0:  96% 95/99 [03:56<00:06,  1.58s/it]\u001b[A\n",
            "Test 0:  97% 96/99 [03:57<00:04,  1.41s/it]\u001b[A\n",
            "Test 0:  98% 97/99 [03:59<00:03,  1.82s/it]\u001b[A\n",
            "Test 0:  99% 98/99 [04:01<00:01,  1.83s/it]\u001b[A\n",
            "Test 0: 100% 99/99 [04:01<00:00,  2.44s/it]\n",
            "test acc =  0.03349206349206349\n",
            "test acc [voted] =  0.03333333333333333\n",
            "11059200it [05:24, 34049.10it/s] \n",
            "\u001b[38;5;2m[i 0703 23:36:29.657766 24 compiler.py:869] Jittor(1.2.3.55) src: /usr/local/lib/python3.7/dist-packages/jittor\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:29.667481 24 compiler.py:870] g++ at /usr/bin/g++(7.5.0)\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:29.667570 24 compiler.py:871] cache_path: /root/.cache/jittor/default/g++\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:29.676098 24 __init__.py:286] Found nvcc(11.0.221) at /usr/local/cuda/bin/nvcc.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:29.684374 24 __init__.py:286] Found addr2line(2.30) at /usr/bin/addr2line.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:29.718414 24 compiler.py:958] py_include: -I/usr/include/python3.7m -I/usr/include/python3.7m\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:29.752147 24 compiler.py:960] extension_suffix: .cpython-37m-x86_64-linux-gnu.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:29.941512 24 compiler.py:1093] OS type:ubuntu OS key:ubuntu\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:29.942388 24 __init__.py:178] Total mem: 25.46GB, using 8 procs for compiling.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:30.212937 24 jit_compiler.cc:22] Load cc_path: /usr/bin/g++\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:30.213699 24 init.cc:55] Found cuda archs: [70,]\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:30.238231 24 __init__.py:286] Found mpicc(2.1.1) at /usr/bin/mpicc.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:30.396579 24 compiler.py:667] handle pyjt_include/usr/local/lib/python3.7/dist-packages/jittor/extern/mpi/inc/mpi_warper.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:30.429092 24 compile_extern.py:347] Downloading nccl...\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:30.492820 24 compile_extern.py:20] found /usr/local/cuda/include/cublas.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:30.503922 24 compile_extern.py:20] found /usr/local/cuda/lib64/libcublas.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:30.504039 24 compile_extern.py:20] found /usr/local/cuda/lib64/libcublasLt.so.11\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:30.928911 24 compile_extern.py:20] found /usr/include/cudnn.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:30.954719 24 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:30.954861 24 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_ops_infer.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:30.957540 24 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_ops_train.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:30.958066 24 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_cnn_infer.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:31.182693 24 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_cnn_train.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:31.197353 24 compiler.py:667] handle pyjt_include/usr/local/lib/python3.7/dist-packages/jittor/extern/cuda/cudnn/inc/cudnn_warper.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:32.393949 24 compile_extern.py:20] found /usr/local/cuda/include/curand.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:32.460667 24 compile_extern.py:20] found /usr/local/cuda/lib64/libcurand.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:36:32.791880 24 cuda_flags.cc:26] CUDA enabled.\u001b[m\n",
            "name:  shrec11-split16\n",
            "Test 0:   0% 0/40 [00:00<?, ?it/s]\n",
            "Compiling Operators(38/38) used: 5.36s eta:    0s \n",
            "Test 0: 100% 40/40 [01:12<00:00,  1.81s/it]\n",
            "test acc =  0.998015873015873\n",
            "test acc [voted] =  1.0\n",
            "\u001b[38;5;2m[i 0703 23:37:52.155491 08 compiler.py:869] Jittor(1.2.3.55) src: /usr/local/lib/python3.7/dist-packages/jittor\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:52.164798 08 compiler.py:870] g++ at /usr/bin/g++(7.5.0)\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:52.164918 08 compiler.py:871] cache_path: /root/.cache/jittor/default/g++\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:52.183956 08 __init__.py:286] Found nvcc(11.0.221) at /usr/local/cuda/bin/nvcc.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:52.292086 08 __init__.py:286] Found addr2line(2.30) at /usr/bin/addr2line.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:52.328846 08 compiler.py:958] py_include: -I/usr/include/python3.7m -I/usr/include/python3.7m\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:52.364398 08 compiler.py:960] extension_suffix: .cpython-37m-x86_64-linux-gnu.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:52.555833 08 compiler.py:1093] OS type:ubuntu OS key:ubuntu\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:52.556681 08 __init__.py:178] Total mem: 25.46GB, using 8 procs for compiling.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:52.823209 08 jit_compiler.cc:22] Load cc_path: /usr/bin/g++\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:52.823916 08 init.cc:55] Found cuda archs: [70,]\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:52.843362 08 __init__.py:286] Found mpicc(2.1.1) at /usr/bin/mpicc.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:52.888484 08 compiler.py:667] handle pyjt_include/usr/local/lib/python3.7/dist-packages/jittor/extern/mpi/inc/mpi_warper.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:52.916840 08 compile_extern.py:347] Downloading nccl...\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:52.976627 08 compile_extern.py:20] found /usr/local/cuda/include/cublas.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:52.987111 08 compile_extern.py:20] found /usr/local/cuda/lib64/libcublas.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:52.987231 08 compile_extern.py:20] found /usr/local/cuda/lib64/libcublasLt.so.11\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:53.413264 08 compile_extern.py:20] found /usr/include/cudnn.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:53.438866 08 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:53.439017 08 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_ops_infer.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:53.441752 08 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_ops_train.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:53.442402 08 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_cnn_infer.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:53.658715 08 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_cnn_train.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:53.672648 08 compiler.py:667] handle pyjt_include/usr/local/lib/python3.7/dist-packages/jittor/extern/cuda/cudnn/inc/cudnn_warper.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:54.837118 08 compile_extern.py:20] found /usr/local/cuda/include/curand.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:54.904991 08 compile_extern.py:20] found /usr/local/cuda/lib64/libcurand.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:37:55.097720 08 cuda_flags.cc:26] CUDA enabled.\u001b[m\n",
            "name:  Cubes\n",
            "Test 0: 100% 103/103 [03:28<00:00,  2.02s/it]\n",
            "test acc =  0.9889226100151745\n",
            "test acc [voted] =  1.0\n",
            "\u001b[38;5;2m[i 0703 23:41:37.404298 48 compiler.py:869] Jittor(1.2.3.55) src: /usr/local/lib/python3.7/dist-packages/jittor\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:37.415913 48 compiler.py:870] g++ at /usr/bin/g++(7.5.0)\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:37.416013 48 compiler.py:871] cache_path: /root/.cache/jittor/default/g++\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:37.426655 48 __init__.py:286] Found nvcc(11.0.221) at /usr/local/cuda/bin/nvcc.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:37.441729 48 __init__.py:286] Found addr2line(2.30) at /usr/bin/addr2line.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:37.478747 48 compiler.py:958] py_include: -I/usr/include/python3.7m -I/usr/include/python3.7m\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:37.513669 48 compiler.py:960] extension_suffix: .cpython-37m-x86_64-linux-gnu.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:37.827892 48 compiler.py:1093] OS type:ubuntu OS key:ubuntu\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:37.829352 48 __init__.py:178] Total mem: 25.46GB, using 8 procs for compiling.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:38.107294 48 jit_compiler.cc:22] Load cc_path: /usr/bin/g++\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:38.108070 48 init.cc:55] Found cuda archs: [70,]\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:38.128508 48 __init__.py:286] Found mpicc(2.1.1) at /usr/bin/mpicc.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:38.179444 48 compiler.py:667] handle pyjt_include/usr/local/lib/python3.7/dist-packages/jittor/extern/mpi/inc/mpi_warper.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:38.212608 48 compile_extern.py:347] Downloading nccl...\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:38.283944 48 compile_extern.py:20] found /usr/local/cuda/include/cublas.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:38.298246 48 compile_extern.py:20] found /usr/local/cuda/lib64/libcublas.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:38.298346 48 compile_extern.py:20] found /usr/local/cuda/lib64/libcublasLt.so.11\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:38.738362 48 compile_extern.py:20] found /usr/include/cudnn.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:38.764375 48 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:38.764499 48 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_ops_infer.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:38.767878 48 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_ops_train.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:38.768551 48 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_cnn_infer.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:41.383478 48 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_cnn_train.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:41.402247 48 compiler.py:667] handle pyjt_include/usr/local/lib/python3.7/dist-packages/jittor/extern/cuda/cudnn/inc/cudnn_warper.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:42.612129 48 compile_extern.py:20] found /usr/local/cuda/include/curand.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:42.682810 48 compile_extern.py:20] found /usr/local/cuda/lib64/libcurand.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:42.902058 48 cuda_flags.cc:26] CUDA enabled.\u001b[m\n",
            "Traceback (most recent call last):\n",
            "  File \"train_cls.py\", line 130, in <module>\n",
            "    shuffle=True, train=True, num_workers=args.n_worker, augment=augments)\n",
            "  File \"/content/drive/My Drive/SubdivNet/subdivnet/dataset.py\", line 110, in __init__\n",
            "    self.browse_dataroot()\n",
            "  File \"/content/drive/My Drive/SubdivNet/subdivnet/dataset.py\", line 116, in browse_dataroot\n",
            "    self.shape_classes = [x.name for x in self.dataroot.iterdir() if x.is_dir()]\n",
            "  File \"/content/drive/My Drive/SubdivNet/subdivnet/dataset.py\", line 116, in <listcomp>\n",
            "    self.shape_classes = [x.name for x in self.dataroot.iterdir() if x.is_dir()]\n",
            "  File \"/usr/lib/python3.7/pathlib.py\", line 1107, in iterdir\n",
            "    for name in self._accessor.listdir(self):\n",
            "FileNotFoundError: [Errno 2] No such file or directory: 'data/Manifold40-MAPS-96-3-0412'\n",
            "\u001b[38;5;2m[i 0703 23:41:46.082816 36 compiler.py:869] Jittor(1.2.3.55) src: /usr/local/lib/python3.7/dist-packages/jittor\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:46.092921 36 compiler.py:870] g++ at /usr/bin/g++(7.5.0)\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:46.093019 36 compiler.py:871] cache_path: /root/.cache/jittor/default/g++\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:46.102722 36 __init__.py:286] Found nvcc(11.0.221) at /usr/local/cuda/bin/nvcc.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:46.111343 36 __init__.py:286] Found addr2line(2.30) at /usr/bin/addr2line.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:46.144577 36 compiler.py:958] py_include: -I/usr/include/python3.7m -I/usr/include/python3.7m\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:46.177987 36 compiler.py:960] extension_suffix: .cpython-37m-x86_64-linux-gnu.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:46.373397 36 compiler.py:1093] OS type:ubuntu OS key:ubuntu\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:46.374391 36 __init__.py:178] Total mem: 25.46GB, using 8 procs for compiling.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:46.672698 36 jit_compiler.cc:22] Load cc_path: /usr/bin/g++\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:46.673459 36 init.cc:55] Found cuda archs: [70,]\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:46.692463 36 __init__.py:286] Found mpicc(2.1.1) at /usr/bin/mpicc.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:46.737902 36 compiler.py:667] handle pyjt_include/usr/local/lib/python3.7/dist-packages/jittor/extern/mpi/inc/mpi_warper.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:46.769348 36 compile_extern.py:347] Downloading nccl...\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:46.835615 36 compile_extern.py:20] found /usr/local/cuda/include/cublas.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:46.848011 36 compile_extern.py:20] found /usr/local/cuda/lib64/libcublas.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:46.848117 36 compile_extern.py:20] found /usr/local/cuda/lib64/libcublasLt.so.11\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:47.283107 36 compile_extern.py:20] found /usr/include/cudnn.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:47.309597 36 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:47.309730 36 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_ops_infer.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:47.312577 36 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_ops_train.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:47.313259 36 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_cnn_infer.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:47.519051 36 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_cnn_train.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:47.534307 36 compiler.py:667] handle pyjt_include/usr/local/lib/python3.7/dist-packages/jittor/extern/cuda/cudnn/inc/cudnn_warper.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:48.706726 36 compile_extern.py:20] found /usr/local/cuda/include/curand.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:48.777195 36 compile_extern.py:20] found /usr/local/cuda/lib64/libcurand.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:41:48.969951 36 cuda_flags.cc:26] CUDA enabled.\u001b[m\n",
            "name: HumanBody\n",
            "0:   0% 0/23 [00:00<?, ?it/s]\n",
            "Compiling Operators(1/1) used: 6.03s eta:    0s \n",
            "\n",
            "Compiling Operators(1/1) used: 3.75s eta:    0s \n",
            "\n",
            "Compiling Operators(1/1) used: 3.65s eta:    0s \n",
            "\n",
            "Compiling Operators(56/56) used: 33.5s eta:    0s \n",
            "0: 100% 23/23 [00:55<00:00,  2.43s/it]\n",
            "test acc =  0.9081288655598958\n",
            "test acc [original] =  0.910494083636563\n",
            "test acc [original] [voted] =  0.9303461568555643\n",
            "test acc per label =  [0.90872471 0.96268618 0.82814069 0.79417724 0.92768654 0.84737467\n",
            " 0.91343925 0.96386798]\n",
            "\u001b[38;5;2m[i 0703 23:43:00.626076 48 compiler.py:869] Jittor(1.2.3.55) src: /usr/local/lib/python3.7/dist-packages/jittor\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:00.635363 48 compiler.py:870] g++ at /usr/bin/g++(7.5.0)\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:00.635446 48 compiler.py:871] cache_path: /root/.cache/jittor/default/g++\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:00.644058 48 __init__.py:286] Found nvcc(11.0.221) at /usr/local/cuda/bin/nvcc.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:00.652636 48 __init__.py:286] Found addr2line(2.30) at /usr/bin/addr2line.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:00.686728 48 compiler.py:958] py_include: -I/usr/include/python3.7m -I/usr/include/python3.7m\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:00.723867 48 compiler.py:960] extension_suffix: .cpython-37m-x86_64-linux-gnu.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:00.924049 48 compiler.py:1093] OS type:ubuntu OS key:ubuntu\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:00.924890 48 __init__.py:178] Total mem: 25.46GB, using 8 procs for compiling.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:01.198831 48 jit_compiler.cc:22] Load cc_path: /usr/bin/g++\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:01.199738 48 init.cc:55] Found cuda archs: [70,]\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:01.221852 48 __init__.py:286] Found mpicc(2.1.1) at /usr/bin/mpicc.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:01.265863 48 compiler.py:667] handle pyjt_include/usr/local/lib/python3.7/dist-packages/jittor/extern/mpi/inc/mpi_warper.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:01.298290 48 compile_extern.py:347] Downloading nccl...\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:01.365480 48 compile_extern.py:20] found /usr/local/cuda/include/cublas.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:01.377857 48 compile_extern.py:20] found /usr/local/cuda/lib64/libcublas.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:01.377951 48 compile_extern.py:20] found /usr/local/cuda/lib64/libcublasLt.so.11\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:01.800491 48 compile_extern.py:20] found /usr/include/cudnn.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:01.826947 48 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:01.827072 48 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_ops_infer.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:01.829923 48 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_ops_train.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:01.830592 48 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_cnn_infer.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:02.055293 48 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_cnn_train.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:02.073060 48 compiler.py:667] handle pyjt_include/usr/local/lib/python3.7/dist-packages/jittor/extern/cuda/cudnn/inc/cudnn_warper.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:03.286590 48 compile_extern.py:20] found /usr/local/cuda/include/curand.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:03.352836 48 compile_extern.py:20] found /usr/local/cuda/lib64/libcurand.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:43:03.544317 48 cuda_flags.cc:26] CUDA enabled.\u001b[m\n",
            "name: coseg-alien\n",
            "0: 100% 37/37 [00:12<00:00,  3.00it/s]\n",
            "test acc =  0.9854860503098061\n",
            "test acc [original] =  0.9638917574330143\n",
            "test acc [original] [voted] =  0.9732876916279151\n",
            "test acc per label =  [0.98391525 0.95060787 0.98919789 0.98310797]\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "dxEKRgTBwtnP",
        "outputId": "4204e3f7-bd39-4ca0-d80f-424e5dc796ae"
      },
      "source": [
        "# !sh scripts/manifold40/test.sh\n",
        "!sh scripts/shrec11-split10/test.sh"
      ],
      "execution_count": 16,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "\u001b[38;5;2m[i 0703 23:45:45.740948 04 compiler.py:869] Jittor(1.2.3.55) src: /usr/local/lib/python3.7/dist-packages/jittor\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:45.752631 04 compiler.py:870] g++ at /usr/bin/g++(7.5.0)\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:45.752716 04 compiler.py:871] cache_path: /root/.cache/jittor/default/g++\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:45.763277 04 __init__.py:286] Found nvcc(11.0.221) at /usr/local/cuda/bin/nvcc.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:45.773713 04 __init__.py:286] Found addr2line(2.30) at /usr/bin/addr2line.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:45.813388 04 compiler.py:958] py_include: -I/usr/include/python3.7m -I/usr/include/python3.7m\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:45.852245 04 compiler.py:960] extension_suffix: .cpython-37m-x86_64-linux-gnu.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:46.064331 04 compiler.py:1093] OS type:ubuntu OS key:ubuntu\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:46.065357 04 __init__.py:178] Total mem: 25.46GB, using 8 procs for compiling.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:46.372999 04 jit_compiler.cc:22] Load cc_path: /usr/bin/g++\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:46.373758 04 init.cc:55] Found cuda archs: [70,]\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:46.396711 04 __init__.py:286] Found mpicc(2.1.1) at /usr/bin/mpicc.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:46.450492 04 compiler.py:667] handle pyjt_include/usr/local/lib/python3.7/dist-packages/jittor/extern/mpi/inc/mpi_warper.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:46.485871 04 compile_extern.py:347] Downloading nccl...\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:46.733014 04 compile_extern.py:20] found /usr/local/cuda/include/cublas.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:46.754249 04 compile_extern.py:20] found /usr/local/cuda/lib64/libcublas.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:46.754372 04 compile_extern.py:20] found /usr/local/cuda/lib64/libcublasLt.so.11\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:47.206048 04 compile_extern.py:20] found /usr/include/cudnn.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:47.232268 04 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:47.232411 04 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_ops_infer.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:47.235465 04 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_ops_train.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:47.236117 04 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_cnn_infer.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:47.457480 04 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_cnn_train.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:47.472945 04 compiler.py:667] handle pyjt_include/usr/local/lib/python3.7/dist-packages/jittor/extern/cuda/cudnn/inc/cudnn_warper.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:48.719985 04 compile_extern.py:20] found /usr/local/cuda/include/curand.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:48.782483 04 compile_extern.py:20] found /usr/local/cuda/lib64/libcurand.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:45:49.154198 04 cuda_flags.cc:26] CUDA enabled.\u001b[m\n",
            "name:  shrec11-split10\n",
            "Test 0: 100% 99/99 [03:13<00:00,  1.96s/it]\n",
            "test acc =  0.03349206349206349\n",
            "test acc [voted] =  0.03333333333333333\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "ZWRwYjxSxzF3",
        "outputId": "cab9ef8f-9caa-4993-f7da-23ec54260afc"
      },
      "source": [
        "!sh scripts/shrec11-split10/train.sh\n"
      ],
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "\u001b[38;5;2m[i 0703 23:50:46.347791 68 compiler.py:869] Jittor(1.2.3.55) src: /usr/local/lib/python3.7/dist-packages/jittor\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:46.359649 68 compiler.py:870] g++ at /usr/bin/g++(7.5.0)\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:46.359784 68 compiler.py:871] cache_path: /root/.cache/jittor/default/g++\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:46.369919 68 __init__.py:286] Found nvcc(11.0.221) at /usr/local/cuda/bin/nvcc.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:46.379842 68 __init__.py:286] Found addr2line(2.30) at /usr/bin/addr2line.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:46.417018 68 compiler.py:958] py_include: -I/usr/include/python3.7m -I/usr/include/python3.7m\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:46.452441 68 compiler.py:960] extension_suffix: .cpython-37m-x86_64-linux-gnu.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:46.642293 68 compiler.py:1093] OS type:ubuntu OS key:ubuntu\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:46.643216 68 __init__.py:178] Total mem: 25.46GB, using 8 procs for compiling.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:46.927512 68 jit_compiler.cc:22] Load cc_path: /usr/bin/g++\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:46.928269 68 init.cc:55] Found cuda archs: [70,]\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:46.948271 68 __init__.py:286] Found mpicc(2.1.1) at /usr/bin/mpicc.\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:46.994946 68 compiler.py:667] handle pyjt_include/usr/local/lib/python3.7/dist-packages/jittor/extern/mpi/inc/mpi_warper.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:47.025383 68 compile_extern.py:347] Downloading nccl...\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:47.089642 68 compile_extern.py:20] found /usr/local/cuda/include/cublas.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:47.101144 68 compile_extern.py:20] found /usr/local/cuda/lib64/libcublas.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:47.101255 68 compile_extern.py:20] found /usr/local/cuda/lib64/libcublasLt.so.11\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:47.519042 68 compile_extern.py:20] found /usr/include/cudnn.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:47.542986 68 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:47.543116 68 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_ops_infer.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:47.545789 68 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_ops_train.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:47.546380 68 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_cnn_infer.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:47.830625 68 compile_extern.py:20] found /usr/lib/x86_64-linux-gnu/libcudnn_cnn_train.so.8\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:47.846399 68 compiler.py:667] handle pyjt_include/usr/local/lib/python3.7/dist-packages/jittor/extern/cuda/cudnn/inc/cudnn_warper.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:49.111908 68 compile_extern.py:20] found /usr/local/cuda/include/curand.h\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:49.172706 68 compile_extern.py:20] found /usr/local/cuda/lib64/libcurand.so\u001b[m\n",
            "\u001b[38;5;2m[i 0703 23:50:49.371886 68 cuda_flags.cc:26] CUDA enabled.\u001b[m\n",
            "name:  shrec11-split10\n",
            "Train 0:   0% 0/99 [00:00<?, ?it/s]\n",
            "Compiling Operators(56/56) used: 28.4s eta:    0s \n",
            "\n",
            "Compiling Operators(43/43) used: 83.6s eta:    0s \n",
            "Train 0:   1% 1/99 [02:02<3:20:18, 122.63s/it]\n",
            "Compiling Operators(2/2) used: 7.32s eta:    0s \n",
            "\n",
            "Compiling Operators(1/1) used: 5.58s eta:    0s \n",
            "Train 0: 100% 99/99 [04:54<00:00,  2.97s/it]\n",
            "train acc =  0.35603174603174603\n",
            "Test 0: 100% 99/99 [03:30<00:00,  2.13s/it]\n",
            "test acc =  0.5406349206349207\n",
            "test acc [voted] =  0.6133333333333333\n",
            "Train 1: 100% 99/99 [03:06<00:00,  1.88s/it]\n",
            "train acc =  0.6955555555555556\n",
            "Test 1: 100% 99/99 [02:58<00:00,  1.81s/it]\n",
            "test acc =  0.5253968253968254\n",
            "test acc [voted] =  0.5766666666666667\n",
            "Train 2: 100% 99/99 [03:03<00:00,  1.85s/it]\n",
            "train acc =  0.833015873015873\n",
            "Test 2:  87% 86/99 [02:36<00:21,  1.63s/it]"
          ],
          "name": "stdout"
        }
      ]
    }
  ]
}