refactored model tests

2026-09-09 11:32:07 +08:00 · 2019-07-24 13:56:21 -04:00
parent cf7da86c7c
commit ef843d5f96
1 changed files with 1 additions and 1 deletions
@@ -111,7 +111,7 @@ def main():
        max_nb_epochs=1,
        gpus=[0, 1],
        distributed_backend='dp',
-        use_amp=True
+        use_amp=False
    )

    result = trainer.fit(model)