lightning/tests/callbacks
Carlos Mocholí f7027a8701
Remove `torch >= 1.6` checks (#8523)
2021-07-23 04:03:20 +00:00
..
__init__.py Progress bar callback (#1450) 2020-04-23 20:46:18 -04:00
test_callback_hook_outputs.py Mark some loop attributes as protected (#8250) 2021-07-02 11:51:51 +01:00
test_callbacks.py Update fit with val hook test (#8060) 2021-06-21 17:27:37 +00:00
test_early_stopping.py Add `ModelCheckpoint(save_on_train_epoch_end)` (#8389) 2021-07-13 14:47:59 +00:00
test_finetuning_callback.py Fix module dict in base finetuning (#8170) 2021-06-28 10:55:32 +00:00
test_gpu_stats_monitor.py Fix: handle logical CUDA device IDs for GPUStatsMonitor if `CUDA_VISIBLE_DEVICES` set (#8260) 2021-07-19 11:42:43 +00:00
test_lambda_function.py Add predict hook test (#7973) 2021-06-16 15:09:24 +02:00
test_lr_monitor.py Remove unnecessary use of comprehension (#8149) 2021-06-27 10:00:02 +01:00
test_prediction_writer.py Remove `trainer.fit` return value [2/n] (#7237) 2021-04-28 19:11:32 +01:00
test_progress_bar.py Support `DDPPlugin` to be used on CPU (#6208) 2021-07-02 12:00:24 +01:00
test_pruning.py [bugfix] Prevent deepcopy of dataloaders / Trainer in SWA Callback (#8472) 2021-07-20 18:31:49 +00:00
test_quantization.py Quant as optional step (#8464) 2021-07-22 12:44:27 +00:00
test_stochastic_weight_avg.py Remove `torch >= 1.6` checks (#8523) 2021-07-23 04:03:20 +00:00
test_timer.py fix info message when max training time reached (#7780) 2021-05-31 14:50:16 +02:00
test_xla_stats_monitor.py Remove unnecessary comprehension (#8405) 2021-07-19 08:30:24 +00:00