We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7890f05 commit 64f680aCopy full SHA for 64f680a
models/rank/wide_deep/criteo_reader.py
@@ -22,7 +22,10 @@ class RecDataset(IterableDataset):
22
def __init__(self, file_list, config):
23
super(RecDataset, self).__init__()
24
self.file_list = file_list
25
- use_fleet = config.get("runner.use_fleet", False)
+ if config:
26
+ use_fleet = config.get("runner.use_fleet", False)
27
+ else:
28
+ use_fleet = False
29
if use_fleet:
30
worker_id = paddle.distributed.get_rank()
31
worker_num = paddle.distributed.get_world_size()
0 commit comments