diff --git a/tests/debug.py b/tests/debug.py index dfa83a0cd9..1375930d91 100644 --- a/tests/debug.py +++ b/tests/debug.py @@ -50,8 +50,8 @@ def main(): progress_bar=True, experiment=get_exp(), max_nb_epochs=1, - train_percent_check=0.4, - val_percent_check=0.4, + train_percent_check=0.1, + val_percent_check=0.1, gpus=[0, 1], distributed_backend='ddp', use_amp=True @@ -62,8 +62,6 @@ def main(): # correct result and ok accuracy assert result == 1, 'amp + ddp model failed to complete' - print(trainer.tng_tqdm_dic) - clear_tt_dir()