@@ -757,11 +757,11 @@ def get_arg_parser():
757
757
help = 'set inter op parallelism threads.' ,
758
758
type = int ,
759
759
default = 0 )
760
- parser .add_argument ('--input_layer_partitioner' , \
760
+ parser .add_argument ('--input_layer_partitioner' ,
761
761
help = 'slice size of input layer partitioner, units MB. Default 8MB' ,
762
762
type = int ,
763
763
default = 8 )
764
- parser .add_argument ('--dense_layer_partitioner' , \
764
+ parser .add_argument ('--dense_layer_partitioner' ,
765
765
help = 'slice size of dense layer partitioner, units KB. Default 16KB' ,
766
766
type = int ,
767
767
default = 16 )
@@ -771,52 +771,52 @@ def get_arg_parser():
771
771
parser .add_argument ('--no_eval' ,
772
772
help = 'not evaluate trained model by eval dataset.' ,
773
773
action = 'store_true' )
774
- parser .add_argument ('--tf' , \
774
+ parser .add_argument ('--tf' ,
775
775
help = 'Use TF 1.15.5 API and disable DeepRec feature to run a baseline.' ,
776
776
action = 'store_true' )
777
- parser .add_argument ('--smartstaged' , \
777
+ parser .add_argument ('--smartstaged' ,
778
778
help = 'Whether to enable smart staged feature of DeepRec, Default to True.' ,
779
779
type = boolean_string ,
780
780
default = True )
781
- parser .add_argument ('--emb_fusion' , \
781
+ parser .add_argument ('--emb_fusion' ,
782
782
help = 'Whether to enable embedding fusion, Default to True.' ,
783
783
type = boolean_string ,
784
784
default = True )
785
- parser .add_argument ('--ev' , \
785
+ parser .add_argument ('--ev' ,
786
786
help = 'Whether to enable DeepRec EmbeddingVariable. Default False.' ,
787
787
type = boolean_string ,
788
788
default = False )
789
- parser .add_argument ('--ev_elimination' , \
789
+ parser .add_argument ('--ev_elimination' ,
790
790
help = 'Feature Elimination of EmbeddingVariable Feature. Default closed.' ,
791
791
type = str ,
792
792
choices = [None , 'l2' , 'gstep' ],
793
793
default = None )
794
- parser .add_argument ('--ev_filter' , \
794
+ parser .add_argument ('--ev_filter' ,
795
795
help = 'Feature Filter of EmbeddingVariable Feature. Default closed.' ,
796
796
type = str ,
797
797
choices = [None , 'counter' , 'cbf' ],
798
798
default = None )
799
- parser .add_argument ('--op_fusion' , \
799
+ parser .add_argument ('--op_fusion' ,
800
800
help = 'Whether to enable Auto graph fusion feature. Default to True' ,
801
801
type = boolean_string ,
802
802
default = True )
803
803
parser .add_argument ('--micro_batch' ,
804
804
help = 'Set num for Auto Mirco Batch. Default close.' ,
805
805
type = int ,
806
806
default = 0 ) #TODO: Defautl to True
807
- parser .add_argument ('--adaptive_emb' , \
807
+ parser .add_argument ('--adaptive_emb' ,
808
808
help = 'Whether to enable Adaptive Embedding. Default to False.' ,
809
809
type = boolean_string ,
810
810
default = False )
811
- parser .add_argument ('--dynamic_ev' , \
811
+ parser .add_argument ('--dynamic_ev' ,
812
812
help = 'Whether to enable Dynamic-dimension Embedding Variable. Default to False.' ,
813
813
type = boolean_string ,
814
814
default = False ) #TODO:enable
815
- parser .add_argument ('--incremental_ckpt' , \
815
+ parser .add_argument ('--incremental_ckpt' ,
816
816
help = 'Set time of save Incremental Checkpoint. Default 0 to close.' ,
817
817
type = int ,
818
818
default = 0 )
819
- parser .add_argument ('--workqueue' , \
819
+ parser .add_argument ('--workqueue' ,
820
820
help = 'Whether to enable Work Queue. Default to False.' ,
821
821
type = boolean_string ,
822
822
default = False )
0 commit comments