FB15k-237_DaBR/notebook.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "provenance": [],
      "gpuType": "T4"
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    },
    "accelerator": "GPU"
  },
  "cells": [
    {
      "cell_type": "code",
      "execution_count": 1,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "B6HE13AWmx1k",
        "outputId": "71365976-6d01-452d-da1b-df37d58aca63"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Cloning into 'DaBR'...\n",
            "remote: Enumerating objects: 99, done.\u001b[K\n",
            "remote: Counting objects: 100% (99/99), done.\u001b[K\n",
            "remote: Compressing objects: 100% (98/98), done.\u001b[K\n",
            "remote: Total 99 (delta 17), reused 0 (delta 0), pack-reused 0 (from 0)\u001b[K\n",
            "Receiving objects: 100% (99/99), 9.47 MiB | 4.77 MiB/s, done.\n",
            "Resolving deltas: 100% (17/17), done.\n",
            "/content/DaBR\n"
          ]
        }
      ],
      "source": [
        "!git clone https://github.com/llqy123/DaBR.git\n",
        "%cd DaBR"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "!python train_FB15K237.py"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "FRQLkAVunBpC",
        "outputId": "9d003b08-7111-4a75-b01b-29b54aa0eb41"
      },
      "execution_count": 2,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Namespace(dataset='FB15K237', learning_rate=0.05, nbatches=100, num_epochs=10000, model_name='DaBR', neg_num=10, hidden_size=500, save_steps=100000, valid_steps=200, lmbda=0.5, lmbda2=0.01, mode='train', checkpoint_path=None, test_file='', optim='adagrad')\n",
            "Writing to /content/DaBR/Logs/FB15K237\n",
            "\n",
            "Input Files Path : ./benchmarks/FB15K237/\n",
            "Test File Path : \n",
            "The toolkit is importing datasets.\n",
            "The total of relations is 237.\n",
            "The total of entities is 14541.\n",
            "The total of train triples is 272115.\n",
            "The total of test triples is 20466.\n",
            "The total of valid triples is 17535.\n",
            "Initializing training model...\n",
            "Finish initializing\n",
            "Epoch 6 | loss: 4.310574:   0% 7/10000 [05:04<120:39:50, 43.47s/it]\n",
            "Traceback (most recent call last):\n",
            "  File \"/content/DaBR/train_FB15K237.py\", line 74, in <module>\n",
            "    con.training_model()\n",
            "  File \"/content/DaBR/config/Config.py\", line 450, in training_model\n",
            "    loss = self.train_one_step()\n",
            "           ^^^^^^^^^^^^^^^^^^^^^\n",
            "  File \"/content/DaBR/config/Config.py\", line 408, in train_one_step\n",
            "    return loss.item()\n",
            "           ^^^^^^^^^^^\n",
            "KeyboardInterrupt\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [],
      "metadata": {
        "id": "waUw4fFUnG8U"
      },
      "execution_count": null,
      "outputs": []
    }
  ]
}