Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
lm-evaluation-harness
Commits
1a183d03
Commit
1a183d03
authored
Oct 20, 2023
by
Michael Pieler
Browse files
fix: copy paste mistake
parent
1964ccd3
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
2 deletions
+4
-2
scripts/write_out.py
scripts/write_out.py
+4
-2
No files found.
scripts/write_out.py
View file @
1a183d03
...
...
@@ -2,6 +2,8 @@ import argparse
import
numpy
as
np
import
json
import
os
import
random
from
lm_eval
import
tasks
from
lm_eval.utils
import
join_iters
from
lm_eval.tasks
import
include_path
from
lm_eval.logger
import
eval_logger
...
...
@@ -39,8 +41,8 @@ def main():
else
:
task_names
=
args
.
tasks
.
split
(
","
)
task_dict
=
tasks
.
get_task_dict
(
task_names
)
os
.
makedirs
(
args
.
output_base_path
,
exist_ok
=
True
)
os
.
makedirs
(
args
.
output_base_path
,
exist_ok
=
True
)
for
task_name
,
task
in
task_dict
.
items
():
rnd
=
random
.
Random
()
rnd
.
seed
(
args
.
seed
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment