Merge branch 'dygraph' into dygraph_for_srn

2026-09-24 23:33:08 +08:00 · 2021-01-26 13:29:13 +08:00
parent ed2f0de95e a27a43ec0f
commit f214437538
129 changed files with 17277 additions and 5070 deletions
@@ -183,8 +183,8 @@ def train(config,
        start_epoch = 1

    for epoch in range(start_epoch, epoch_num + 1):
-        if epoch > 0:
-            train_dataloader = build_dataloader(config, 'Train', device, logger)
+        train_dataloader = build_dataloader(
+            config, 'Train', device, logger, seed=epoch)
        train_batch_cost = 0.0
        train_reader_cost = 0.0
        batch_sum = 0