@@ -873,11 +873,11 @@ def get_arg_parser():
873
873
help = 'set inter op parallelism threads.' ,
874
874
type = int ,
875
875
default = 0 )
876
- parser .add_argument ('--input_layer_partitioner' , \
876
+ parser .add_argument ('--input_layer_partitioner' ,
877
877
help = 'slice size of input layer partitioner, units MB. Default 8MB' ,
878
878
type = int ,
879
879
default = 8 )
880
- parser .add_argument ('--dense_layer_partitioner' , \
880
+ parser .add_argument ('--dense_layer_partitioner' ,
881
881
help = 'slice size of dense layer partitioner, units KB. Default 16KB' ,
882
882
type = int ,
883
883
default = 16 )
@@ -887,52 +887,52 @@ def get_arg_parser():
887
887
parser .add_argument ('--no_eval' ,
888
888
help = 'not evaluate trained model by eval dataset.' ,
889
889
action = 'store_true' )
890
- parser .add_argument ('--tf' , \
890
+ parser .add_argument ('--tf' ,
891
891
help = 'Use TF 1.15.5 API and disable DeepRec feature to run a baseline.' ,
892
892
action = 'store_true' )
893
- parser .add_argument ('--smartstaged' , \
893
+ parser .add_argument ('--smartstaged' ,
894
894
help = 'Whether to enable smart staged feature of DeepRec, Default to True.' ,
895
895
type = boolean_string ,
896
896
default = True )
897
- parser .add_argument ('--emb_fusion' , \
897
+ parser .add_argument ('--emb_fusion' ,
898
898
help = 'Whether to enable embedding fusion, Default to True.' ,
899
899
type = boolean_string ,
900
900
default = True )
901
- parser .add_argument ('--ev' , \
901
+ parser .add_argument ('--ev' ,
902
902
help = 'Whether to enable DeepRec EmbeddingVariable. Default False.' ,
903
903
type = boolean_string ,
904
904
default = False )
905
- parser .add_argument ('--ev_elimination' , \
905
+ parser .add_argument ('--ev_elimination' ,
906
906
help = 'Feature Elimination of EmbeddingVariable Feature. Default closed.' ,
907
907
type = str ,
908
908
choices = [None , 'l2' , 'gstep' ],
909
909
default = None )
910
- parser .add_argument ('--ev_filter' , \
910
+ parser .add_argument ('--ev_filter' ,
911
911
help = 'Feature Filter of EmbeddingVariable Feature. Default closed.' ,
912
912
type = str ,
913
913
choices = [None , 'counter' , 'cbf' ],
914
914
default = None )
915
- parser .add_argument ('--op_fusion' , \
915
+ parser .add_argument ('--op_fusion' ,
916
916
help = 'Whether to enable Auto graph fusion feature. Default to True' ,
917
917
type = boolean_string ,
918
918
default = True )
919
919
parser .add_argument ('--micro_batch' ,
920
920
help = 'Set num for Auto Mirco Batch. Default close.' ,
921
921
type = int ,
922
922
default = 0 ) #TODO: Defautl to True
923
- parser .add_argument ('--adaptive_emb' , \
923
+ parser .add_argument ('--adaptive_emb' ,
924
924
help = 'Whether to enable Adaptive Embedding. Default to False.' ,
925
925
type = boolean_string ,
926
926
default = False )
927
- parser .add_argument ('--dynamic_ev' , \
927
+ parser .add_argument ('--dynamic_ev' ,
928
928
help = 'Whether to enable Dynamic-dimension Embedding Variable. Default to False.' ,
929
929
type = boolean_string ,
930
930
default = False ) #TODO:enable
931
- parser .add_argument ('--incremental_ckpt' , \
931
+ parser .add_argument ('--incremental_ckpt' ,
932
932
help = 'Set time of save Incremental Checkpoint. Default 0 to close.' ,
933
933
type = int ,
934
934
default = 0 )
935
- parser .add_argument ('--workqueue' , \
935
+ parser .add_argument ('--workqueue' ,
936
936
help = 'Whether to enable Work Queue. Default to False.' ,
937
937
type = boolean_string ,
938
938
default = False )
0 commit comments