remove summary

2026-09-11 12:31:23 +08:00 · 2019-12-11 07:27:32 -08:00 · 2019-12-11 06:50:56 -08:00 · 2019-12-11 06:50:44 -08:00 · 2019-12-11 06:50:20 -08:00 · 2019-12-11 06:48:30 -08:00
2 changed files with 3 additions and 1 deletions
@@ -47,7 +47,7 @@ def test_no_amp_single_gpu(tmpdir):
        show_progress_bar=True,
        max_epochs=1,
        gpus=1,
-        distributed_backend='dp',
+        distributed_backend=None,
        use_amp=True
    )

@@ -144,6 +144,8 @@ def load_model(exp, root_weights_dir, module_class=LightningTemplateModel):


 def run_prediction(dataloader, trained_model, dp=False, min_acc=0.50):
+    trained_model.eval()
+
    # run prediction on 1 batch
    for batch in dataloader:
        break
Author	SHA1	Message	Date
William Falcon	397c0754d8	remove summary	2019-12-11 07:27:32 -08:00
William Falcon	486a841dab	remove summary	2019-12-11 06:50:56 -08:00
William Falcon	28b68503bc	remove summary	2019-12-11 06:50:44 -08:00
William Falcon	23f1b98c0e	remove summary	2019-12-11 06:50:20 -08:00
William Falcon	bb3c934805	remove summary	2019-12-11 06:48:30 -08:00
William Falcon	44de0b3563	remove summary	2019-12-11 06:44:20 -08:00
William Falcon	47bdc77bce	remove summary	2019-12-11 06:31:04 -08:00
William Falcon	6c5a6a1b4d	remove summary	2019-12-11 06:26:12 -08:00
William Falcon	84c23d3a1d	remove summary	2019-12-11 06:23:21 -08:00
William Falcon	e8fb2fc111	remove summary	2019-12-11 06:21:36 -08:00