removed dummy d

2026-09-09 11:32:07 +08:00 · 2019-07-24 11:00:36 -04:00
parent 8e131f9d79
commit 5606fd86df
1 changed files with 1 additions and 1 deletions
@@ -63,7 +63,7 @@ def main():
    assert result == 1, 'amp + ddp model failed to complete'

    # test prediction
-    data = model.test_dataloader
+    data = model.val_dataloader
    for batch in data:
        break