InternLM/InternBootcamp
1
0
Fork 0
mirror of https://github.com/InternLM/InternBootcamp.git synced 2026-04-19 12:58:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

pjli-dev

Browse source
This commit is contained in:
lipeiji 2025-06-12 12:45:31 +08:00
parent e5d5e53728
commit b379c541bf
11 changed files with 239 additions and 223 deletions
Download patch file Download diff file Expand all files Collapse all files

2
examples/pipelines/cipher_data_generator.py
View file

@ -170,12 +170,14 @@ if __name__ == '__main__':
parser.add_argument('--split', type=str)
parser.add_argument('--timestamp', type=str)
parser.add_argument('--filepath', type=str)
parser.add_argument('--coef', type=float, default=30/144, help='Coef for balance cipher data tasks')
args = parser.parse_args()
# 获取传入的参数
nums = int(args.nums)
if nums == 0:
exit()
nums = int(nums * args.coef)
split = str(args.split)
timestamp = str(args.timestamp)
# df = pd.read_csv(r'internbootcamp/libs/data/train.tsv', sep='\t')