Skip to content

Commit 308a394

Browse files
author
root
committed
format
1 parent 8daf1a2 commit 308a394

File tree

1 file changed

+5
-12
lines changed

1 file changed

+5
-12
lines changed

python/paddle/fluid/dataset.py

Lines changed: 5 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -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

10631064
class 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

Comments
 (0)