@@ -1041,7 +1041,7 @@ def _set_heter_ps(self, enable_heter_ps=False):
10411041 user no need to call this function.
10421042 """
10431043 self .dataset .set_heter_ps (enable_heter_ps )
1044-
1044+
10451045 def set_graph_device_keys (self , device_keys ):
10461046 """
10471047 """
@@ -1053,11 +1053,12 @@ def set_graph_config(self, config):
10531053 self .proto_desc .graph_config .walk_degree = config .get ("walk_degree" , 1 )
10541054 self .proto_desc .graph_config .walk_len = config .get ("walk_len" , 20 )
10551055 self .proto_desc .graph_config .window = config .get ("window" , 5 )
1056- self .proto_desc .graph_config .once_sample_startid_len = config .get ("once_sample_startid_len" , 8000 )
1057- self .proto_desc .graph_config .sample_times_one_chunk = config .get ("sample_times_one_chunk" , 10 )
1056+ self .proto_desc .graph_config .once_sample_startid_len = config .get (
1057+ "once_sample_startid_len" , 8000 )
1058+ self .proto_desc .graph_config .sample_times_one_chunk = config .get (
1059+ "sample_times_one_chunk" , 10 )
10581060 self .proto_desc .graph_config .batch_size = config .get ("batch_size" , 1 )
10591061 self .proto_desc .graph_config .debug_mode = config .get ("debug_mode" , 0 )
1060-
10611062
10621063
10631064class QueueDataset (DatasetBase ):
@@ -1294,14 +1295,6 @@ def _dynamic_adjust_before_train(self, thread_num):
12941295 def _dynamic_adjust_after_train (self ):
12951296 pass
12961297
1297-
1298-
1299-
1300-
1301-
1302-
1303-
1304-
13051298 def slots_shuffle (self , slots ):
13061299 """
13071300 Slots Shuffle
0 commit comments