Skip to content

Commit abf387c

Browse files
add train_parallel
1 parent 89453a7 commit abf387c

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

deep_keyphrase/copy_rnn/train.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -172,6 +172,7 @@ def parse_args(self):
172172
parser.add_argument('-logfile', type=str, default='train_log.log', help='')
173173
parser.add_argument('-save_model_step', type=int, default=5000, help='')
174174
parser.add_argument('-early_stop_tolerance', type=int, default=100, help='')
175+
parser.add_argument('-train_parallel', action='store_true', help='')
175176
parser.add_argument('-schedule_lr', action='store_true', help='')
176177
parser.add_argument('-schedule_step', type=int, default=100000, help='')
177178
parser.add_argument('-schedule_gamma', type=float, default=0.5, help='')

0 commit comments

Comments
 (0)