Skip to content

Commit

Permalink
remove unnecessary slashes
Browse files Browse the repository at this point in the history
Signed-off-by: co63oc <[email protected]>
  • Loading branch information
co63oc committed Apr 18, 2023
1 parent 3bc930a commit 8c74329
Showing 1 changed file with 13 additions and 13 deletions.
26 changes: 13 additions & 13 deletions modelzoo/bst/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -873,11 +873,11 @@ def get_arg_parser():
help='set inter op parallelism threads.',
type=int,
default=0)
parser.add_argument('--input_layer_partitioner', \
parser.add_argument('--input_layer_partitioner',
help='slice size of input layer partitioner, units MB. Default 8MB',
type=int,
default=8)
parser.add_argument('--dense_layer_partitioner', \
parser.add_argument('--dense_layer_partitioner',
help='slice size of dense layer partitioner, units KB. Default 16KB',
type=int,
default=16)
Expand All @@ -887,52 +887,52 @@ def get_arg_parser():
parser.add_argument('--no_eval',
help='not evaluate trained model by eval dataset.',
action='store_true')
parser.add_argument('--tf', \
parser.add_argument('--tf',
help='Use TF 1.15.5 API and disable DeepRec feature to run a baseline.',
action='store_true')
parser.add_argument('--smartstaged', \
parser.add_argument('--smartstaged',
help='Whether to enable smart staged feature of DeepRec, Default to True.',
type=boolean_string,
default=True)
parser.add_argument('--emb_fusion', \
parser.add_argument('--emb_fusion',
help='Whether to enable embedding fusion, Default to True.',
type=boolean_string,
default=True)
parser.add_argument('--ev', \
parser.add_argument('--ev',
help='Whether to enable DeepRec EmbeddingVariable. Default False.',
type=boolean_string,
default=False)
parser.add_argument('--ev_elimination', \
parser.add_argument('--ev_elimination',
help='Feature Elimination of EmbeddingVariable Feature. Default closed.',
type=str,
choices=[None, 'l2', 'gstep'],
default=None)
parser.add_argument('--ev_filter', \
parser.add_argument('--ev_filter',
help='Feature Filter of EmbeddingVariable Feature. Default closed.',
type=str,
choices=[None, 'counter', 'cbf'],
default=None)
parser.add_argument('--op_fusion', \
parser.add_argument('--op_fusion',
help='Whether to enable Auto graph fusion feature. Default to True',
type=boolean_string,
default=True)
parser.add_argument('--micro_batch',
help='Set num for Auto Mirco Batch. Default close.',
type=int,
default=0) #TODO: Defautl to True
parser.add_argument('--adaptive_emb', \
parser.add_argument('--adaptive_emb',
help='Whether to enable Adaptive Embedding. Default to False.',
type=boolean_string,
default=False)
parser.add_argument('--dynamic_ev', \
parser.add_argument('--dynamic_ev',
help='Whether to enable Dynamic-dimension Embedding Variable. Default to False.',
type=boolean_string,
default=False) #TODO:enable
parser.add_argument('--incremental_ckpt', \
parser.add_argument('--incremental_ckpt',
help='Set time of save Incremental Checkpoint. Default 0 to close.',
type=int,
default=0)
parser.add_argument('--workqueue', \
parser.add_argument('--workqueue',
help='Whether to enable Work Queue. Default to False.',
type=boolean_string,
default=False)
Expand Down

0 comments on commit 8c74329

Please sign in to comment.