Skip to content

Commit fa786cc

Browse files
authored
remove hard coded input from category file generation (#4841)
1 parent 0817f7f commit fa786cc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchvision/prototype/datasets/generate_category_files.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def parse_args(argv=None):
5252

5353

5454
if __name__ == "__main__":
55-
args = parse_args(["-f", "sbd"])
55+
args = parse_args()
5656

5757
try:
5858
main(*args.names, force=args.force)

0 commit comments

Comments
 (0)