renamed project output

Author: codinglabsong

src/bart_reddit_lora/model.py

@@ -34,7 +34,7 @@ def build_peft_model(
 
 def load_peft_model_for_inference(
     base_model: BartForConditionalGeneration,
-    adapter_path: str = "outputs/bart-base-korea-travel-guide-lora",
+    adapter_path: str = "outputs/bart-base-reddit-lora",
 ) -> PeftModel:
     inference_model = PeftModel.from_pretrained(base_model, adapter_path).eval()
     return inference_model

src/bart_reddit_lora/train.py

@@ -43,7 +43,7 @@ class DataArgs:
 class CustomTrainingArgs(Seq2SeqTrainingArguments):
     # overriding the hf defaults
     output_dir: str = field(
-        default="outputs/bart-base-korea-travel-guide-lora",
+        default="outputs/bart-base-reddit-lora",
         metadata={"help": "Prefix folder for all checkpoints/run logs."},
     )
     num_train_epochs: int = 6

src/bart_reddit_lora/utils.py

@@ -26,7 +26,7 @@ def print_trainable_parameters(model: PreTrainedModel) -> None:
     return f"trainable params: {trainable_params} || all params: {all_param} || trainable%: {100 * trainable_params / all_param:.2f}"
 
 
-def load_environ_vars(wandb_project: str = "bart-base-korea-travel-guide-lora"):
+def load_environ_vars(wandb_project: str = "bart-base-reddit-lora"):
     # hf hub
     token = os.getenv("HUGGINGFACE_TOKEN")
     if token:

sweep.yaml

@@ -1,7 +1,7 @@
-# sweep spec for Korea-Travel-Guide BART
+# sweep spec for bart-base-reddit-lora
 program: scripts/train.py
 
-project: bart-base-korea-travel-guide-lora
+project: bart-base-reddit-lora
 entity:  codinglabsong-keio-jp
 
 method: bayes # {grid | random | bayes}