Skip to content

Commit e5d16bd

Browse files
Update main.py
1 parent 5df1f04 commit e5d16bd

File tree

1 file changed

+17
-21
lines changed

1 file changed

+17
-21
lines changed

main.py

Lines changed: 17 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -53,27 +53,23 @@ def parse_args():
5353
parser.add_argument('--amp',
5454
action='store_true',
5555
help='whether to open amp training.')
56-
parser.add_argument(
57-
'--validate',
58-
action='store_true',
59-
help='whether to evaluate the checkpoint during training')
60-
parser.add_argument(
61-
'--seed',
62-
type=int,
63-
default=None,
64-
help='fixed all random seeds when the program is running')
65-
parser.add_argument(
66-
'--max_iters',
67-
type=int,
68-
default=None,
69-
help='max iterations when training(this argonly used in test_tipc)')
70-
parser.add_argument(
71-
'-p',
72-
'--profiler_options',
73-
type=str,
74-
default=None,
75-
help='The option of profiler, which should be in format '
76-
'\"key1=value1;key2=value2;key3=value3\".')
56+
parser.add_argument('--validate',
57+
action='store_true',
58+
help='whether to evaluate the checkpoint during training')
59+
parser.add_argument('--seed',
60+
type=int,
61+
default=None,
62+
help='fixed all random seeds when the program is running')
63+
parser.add_argument('--max_iters',
64+
type=int,
65+
default=None,
66+
help='max iterations when training(this argonly used in test_tipc)')
67+
parser.add_argument('-p',
68+
'--profiler_options',
69+
type=str,
70+
default=None,
71+
help='The option of profiler, which should be in format '
72+
'\"key1=value1;key2=value2;key3=value3\".')
7773

7874
args = parser.parse_args()
7975
return args

0 commit comments

Comments
 (0)