removed dummy d

2026-09-08 17:11:08 +08:00 · 2019-07-24 10:55:17 -04:00
parent e4313b0b3d
commit b684fdf502
1 changed files with 8 additions and 0 deletions
@@ -62,6 +62,14 @@ def main():
    # correct result and ok accuracy
    assert result == 1, 'amp + ddp model failed to complete'

+    # test prediction
+    data = model.test_dataloader
+    for batch in data:
+        break
+    out = model(data[0])
+    print(out)
+
+
    clear_tt_dir()