[#33]FIX: global.yaml 수정

KDT-AiVENGERS · Jul 29, 2023 · 2a90a9c · 2a90a9c
1 parent 24e0373
commit 2a90a9c
Show file tree

Hide file tree

Showing 7 changed files with 32 additions and 48 deletions.
diff --git a/experiment/ch/global.yaml b/experiment/ch/global.yaml
@@ -1,2 +1,3 @@
 next_sweep_count: 0
 next_version_count: 17
+next_hugging_face_count: 0
diff --git a/experiment/ej/ckpt2huggingface.ipynb b/experiment/ej/ckpt2huggingface.ipynb
@@ -67,7 +67,7 @@
     "import ruamel.yaml\n",
     "import wandb\n",
     "import warnings\n",
-    "warnings.filterwarnings(\"ignore\", category=UserWarning)\n"
+    "warnings.filterwarnings(\"ignore\", category=UserWarning)"
    ]
   },
   {
@@ -97,7 +97,7 @@
     "        for key in self.keys:\n",
     "            item.append(self.data[key][index])\n",
     "\n",
-    "        return item"
+    "        return item\n"
    ]
   },
   {
@@ -292,7 +292,7 @@
     "        return DataLoader(self.test_dataset, batch_size=self.batch_size, collate_fn=self._collate_fn_train)\n",
     "\n",
     "    def predict_dataloader(self) -> EVAL_DATALOADERS:\n",
-    "        return DataLoader(self.predict_dataset, batch_size=self.predict_batch_size, collate_fn=self._collate_fn_predict)\n"
+    "        return DataLoader(self.predict_dataset, batch_size=self.predict_batch_size, collate_fn=self._collate_fn_predict)"
    ]
   },
   {
@@ -430,12 +430,12 @@
     "        if self.lr_scheduler is not None:\n",
     "            return [optimizer], [self.lr_scheduler]\n",
     "        else:\n",
-    "            return optimizer\n"
+    "            return optimizer"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 14,
+   "execution_count": 15,
    "metadata": {},
    "outputs": [
     {
@@ -484,7 +484,7 @@
     "                     allow_unicode=True)\n",
     "\n",
     "\n",
-    "hf_trainer.save_model(save_dir)"
+    "hf_trainer.save_model(save_dir)\n"
    ]
   }
  ],

diff --git a/experiment/ej/predict.ipynb b/experiment/ej/predict.ipynb
diff --git a/experiment/ih/global.yaml b/experiment/ih/global.yaml
@@ -1,2 +1,3 @@
 next_sweep_count: 0
 next_version_count: 17
+next_hugging_face_count: 0
diff --git a/experiment/jh/global.yaml b/experiment/jh/global.yaml
@@ -1,2 +1,3 @@
 next_sweep_count: 0
 next_version_count: 17
+next_hugging_face_count: 0
diff --git a/experiment/sl/global.yaml b/experiment/sl/global.yaml
@@ -1,2 +1,3 @@
 next_sweep_count: 0
 next_version_count: 17
+next_hugging_face_count: 0
diff --git a/experiment/tg/global.yaml b/experiment/tg/global.yaml
@@ -1,2 +1,3 @@
 next_sweep_count: 0
 next_version_count: 17
+next_hugging_face_count: 0