nicholasKluge
/

Aira-OPT-125M

Text Generation

instruction tuned

text generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

nicholasKluge commited on Oct 30, 2023

Commit

cab1c48

•

1 Parent(s): a8a8d39

Upload AIRA_FineTuning.ipynb

Files changed (1) hide show

AIRA_FineTuning.ipynb +1 -1

AIRA_FineTuning.ipynb CHANGED Viewed

@@ -321,7 +321,7 @@
         "dataloader = DataLoader(\n",
         "            dataset,\n",
         "            sampler=RandomSampler(dataset),\n",
-        "            batch_size=4\n",
         "        )"
       ]
     },

         "dataloader = DataLoader(\n",
         "            dataset,\n",
         "            sampler=RandomSampler(dataset),\n",
+        "            batch_size=32\n",
         "        )"
       ]
     },