fixed amp bug

2026-09-11 12:31:23 +08:00 · 2019-07-24 14:11:05 -04:00
parent dd4f8899c8
commit 1b273a32ee
2 changed files with 9 additions and 1 deletions
@@ -111,7 +111,7 @@ def main():
        max_nb_epochs=1,
        gpus=[0, 1],
        distributed_backend='dp',
-        use_amp=False
+        use_amp=True
    )

    result = trainer.fit(model)