Skip to content

Commit 00c488d

Browse files
committed
add ignore center
1 parent 3208600 commit 00c488d

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

train.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -367,6 +367,8 @@ def run(args, both_cases=False):
367367
parser.add_argument('--steps', nargs='+', default=['lower'])
368368
parser.add_argument('--group_labels', type=int, default=1, choices=[0, 1])
369369
parser.add_argument('--ignore_center', type=int, default=1, choices=[0, 1])
370+
parser.add_argument('--label_ignored', type=int, default=None)
371+
370372
parser.add_argument('--ratio', type=float, default=1)
371373
parser.add_argument('--balance', type=int, default=0, choices=[0, 1])
372374
parser.add_argument('--use_sampler', type=int,

0 commit comments

Comments
 (0)