Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/llama-factory.git" did not exist on "e92143e3fc53b0cef786700b2cceab5564a3367f"
Commit 3f58649b authored by Mohammad's avatar Mohammad
Browse files

utils partially refactored and tested

parent 86e7d624
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tasks/race/data.py
View file @ 3f58649b
...@@ -6,7 +6,7 @@ import time ...@@ -6,7 +6,7 @@ import time
from torch.utils.data import Dataset from torch.utils.data import Dataset
from megatron.utils import print_rank_0 from megatron import print_rank_0
from tasks.data_utils import build_sample from tasks.data_utils import build_sample
from tasks.data_utils import build_tokens_types_paddings_from_ids from tasks.data_utils import build_tokens_types_paddings_from_ids
from tasks.data_utils import clean_text from tasks.data_utils import clean_text
......
tasks/race/finetune.py
View file @ 3f58649b
"""Race.""" """Race."""
from megatron.model.multiple_choice import MultipleChoice from megatron.model.multiple_choice import MultipleChoice
from megatron.utils import print_rank_0 from megatron import print_rank_0
from tasks.eval_utils import accuracy_func_provider from tasks.eval_utils import accuracy_func_provider
from tasks.finetune_utils import finetune from tasks.finetune_utils import finetune
from tasks.race.data import RaceDataset from tasks.race.data import RaceDataset
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment