Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Modelzoo] Remove unnecessary slashes #510

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
Open

[Modelzoo] Remove unnecessary slashes #510

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
8c74329
remove unnecessary slashes
co63oc Apr 18, 2023
6ad194d
Merge branch 'DeepRec-AI:main' into main
co63oc May 15, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
26 changes: 13 additions & 13 deletions modelzoo/bst/train.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -875,11 +875,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 @@ -889,52 +889,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