Skip to content

Commit 64f680a

Browse files
committed
fix reader has no config
1 parent 7890f05 commit 64f680a

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

models/rank/wide_deep/criteo_reader.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,10 @@ class RecDataset(IterableDataset):
2222
def __init__(self, file_list, config):
2323
super(RecDataset, self).__init__()
2424
self.file_list = file_list
25-
use_fleet = config.get("runner.use_fleet", False)
25+
if config:
26+
use_fleet = config.get("runner.use_fleet", False)
27+
else:
28+
use_fleet = False
2629
if use_fleet:
2730
worker_id = paddle.distributed.get_rank()
2831
worker_num = paddle.distributed.get_world_size()

0 commit comments

Comments
 (0)