Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Rename training utils filenames to be easily understandable
Browse files Browse the repository at this point in the history
SachiraKuruppu committed Sep 17, 2023
1 parent 3075244 commit 11f2637
Showing 4 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion dalm/training/rag_e2e/train_rage2e.py
Original file line number Diff line number Diff line change
@@ -42,7 +42,7 @@
)

from dalm.models.rag_e2e_base_model import AutoModelForRagE2E
from dalm.training.utils.rag_e2e_training_utils import preprocess_dataset
from dalm.training.utils.rag_e2e_dataloader_utils import preprocess_dataset
from dalm.training.utils.train_utils import (
compute_marginalized_loss_from_logits,
get_cosine_sim,
2 changes: 1 addition & 1 deletion dalm/training/retriever_only/train_retriever_only.py
Original file line number Diff line number Diff line change
@@ -33,7 +33,7 @@
from transformers import AutoTokenizer, SchedulerType, default_data_collator, get_scheduler

from dalm.models.retriever_only_base_model import AutoModelForSentenceEmbedding
from dalm.training.utils.retriever_only_training_utils import preprocess_dataset
from dalm.training.utils.retriever_only_dataloader_utils import preprocess_dataset
from dalm.training.utils.train_utils import get_cosine_sim, get_nt_xent_loss, load_model_hook, save_model_hook

logger = get_logger(__name__)

0 comments on commit 11f2637

Please sign in to comment.