diff --git a/tests/debug.py b/tests/debug.py index 531201fa88..2833b651a5 100644 --- a/tests/debug.py +++ b/tests/debug.py @@ -111,7 +111,6 @@ def main(): max_nb_epochs=1, gpus=[0, 1], distributed_backend='dp', - use_amp=True ) result = trainer.fit(model)