added sample input for summary

2026-09-11 12:31:23 +08:00 · 2019-07-24 16:35:32 -04:00
parent b824f184ff
commit 383b4cdac7
1 changed files with 1 additions and 1 deletions
@@ -137,7 +137,7 @@ def test_amp_gpu_dp():

    trainer_options = dict(
        max_nb_epochs=1,
-        gpus=[0, 1],
+        gpus='0, 1',  # test init with gpu string
        distributed_backend='dp',
        use_amp=True
    )