Skip to content

GitLab

Unverified Commit f6e7ae25 authored by Leo Gao's avatar Leo Gao Committed by GitHub
Browse files

Update main.py

parent af0b1e26
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
main.py
View file @ f6e7ae25
...@@ -16,7 +16,7 @@ def parse_args(): ...@@ -16,7 +16,7 @@ def parse_args():
parser.add_argument('--provide_description', action="store_true") parser.add_argument('--provide_description', action="store_true")
parser.add_argument('--num_fewshot', type=int, default=0) parser.add_argument('--num_fewshot', type=int, default=0)
parser.add_argument('--batch_size', type=int, default=None) parser.add_argument('--batch_size', type=int, default=None)
parser.add_argument('--device', type=int, default=None) parser.add_argument('--device', type=str, default=None)
parser.add_argument('--seed', type=int, default=1234) parser.add_argument('--seed', type=int, default=1234)
parser.add_argument('--output_path', default=None) parser.add_argument('--output_path', default=None)
parser.add_argument('--limit', type=int, default=None) parser.add_argument('--limit', type=int, default=None)
......
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