Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

update dataset preprocess #1257

Merged
merged 5 commits into from
Jun 30, 2024

Conversation

Jintao-Huang
Copy link
Collaborator

@Jintao-Huang Jintao-Huang commented Jun 29, 2024

PR type

  • Bug Fix
  • New Feature
  • Document Updates
  • More Models or Datasets Support
  1. 重构dataset, preprocess部分代码和逻辑 (注意:未对ListPreprocessor进行修改)
from swift.llm import get_dataset

dataset = get_dataset('iic/ms_bench')[0]
"""
# 优化前
Map: 100%|███████████████████████████████████████████████| 317623/317623 [00:31<00:00, 10085.15 examples/s]
Filter: 100%|████████████████████████████████████████████| 317623/317623 [00:10<00:00, 30777.00 examples/s]
Map: 100%|████████████████████████████████████████████████| 316734/316734 [00:46<00:00, 6880.82 examples/s]

# 优化后
Map: 100%|█████████████████████████████████████████████████| 317623/317623 [00:31<00:00, 9993.46 examples/s]
Filter: 100%|█████████████████████████████████████████████| 317623/317623 [00:10<00:00, 31072.59 examples/s]
"""

@Jintao-Huang Jintao-Huang merged commit 014a21f into modelscope:main Jun 30, 2024
2 checks passed
hjh0119 pushed a commit to hjh0119/swift that referenced this pull request Jul 22, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants