updated colab notebook

hkproj · May 22, 2023 · fb67db8 · fb67db8
1 parent 3f2658e
commit fb67db8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Colab_Train.ipynb b/Colab_Train.ipynb
@@ -20,7 +20,6 @@
       "outputs": [],
       "source": [
         "%%capture\n",
-        "!echo \"Installing requirements\"\n",
         "!pip install datasets\n",
         "!pip install tokenizers\n",
         "!pip install torchmetrics"
@@ -288,7 +287,8 @@
         "cfg = get_config()\n",
         "cfg['model_folder'] = '..//drive/MyDrive/Models/pytorch-transformer/weights'\n",
         "cfg['tokenizer_file'] = '..//drive/MyDrive/Models/pytorch-transformer/vocab/tokenizer_{0}.json'\n",
-        "cfg['batch_size'] = 16\n",
+        "cfg['batch_size'] = 24\n",
+        "cfg['num_epochs'] = 100\n",
         "cfg['preload'] = None\n",
         "\n",
         "from train import train_model\n",