Merge pull request #273 from lakshith-403/LoRA

LoRA rename configs to trainer
labmlai · Aug 24, 2024 · 90e21b5 · 90e21b5
2 parents 6df1d79 + 309a071
commit 90e21b5
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/labml_nn/lora/experiment.ipynb b/labml_nn/lora/experiment.ipynb
@@ -1,10 +1,18 @@
 {
  "cells": [
+  {
+   "metadata": {},
+   "cell_type": "code",
+   "outputs": [],
+   "execution_count": null,
+   "source": "!pip install labml-nn",
+   "id": "c5ed37230628ee76"
+  },
   {
    "metadata": {},
    "cell_type": "code",
    "source": [
-    "from labml_nn.lora.experiment import Configs\n",
+    "from labml_nn.lora.experiment import Trainer\n",
     "from labml import experiment"
    ],
    "id": "1b9da2e59ffce5d5",
@@ -24,23 +32,23 @@
   {
    "metadata": {},
    "cell_type": "code",
-   "source": "conf = Configs()",
+   "source": "trainer = Trainer()",
    "id": "31c9bc08eca2592",
    "outputs": [],
    "execution_count": null
   },
   {
    "metadata": {},
    "cell_type": "code",
-   "source": "experiment.configs(conf)",
+   "source": "experiment.configs(trainer)",
    "id": "fb6ce74326558948",
    "outputs": [],
    "execution_count": null
   },
   {
    "metadata": {},
    "cell_type": "code",
-   "source": "conf.initialize()",
+   "source": "trainer.initialize()",
    "id": "1456cfab47dee3b",
    "outputs": [],
    "execution_count": null
@@ -50,7 +58,7 @@
    "cell_type": "code",
    "source": [
     "with experiment.start():\n",
-    "    conf.run()"
+    "    trainer.run()"
    ],
    "id": "3fe4068fd2df9094",
    "outputs": [],