lightning/tests/trainer
Carlos Mocholí 0dd6d3a798
Avoid adding `None` loss values in `training_epoch_end` (#7772)
2021-05-31 19:28:28 +00:00
..
connectors Add checkpoint parameter to on_save_checkpoint (#6072) 2021-02-25 21:18:19 +05:30
dynamic_args
flags Remove metric tracking from dev debugger (#7759) 2021-05-30 12:03:42 +02:00
logging_ Clean existing logging tests (#7760) 2021-05-30 16:36:52 +02:00
loops Add Test for memory consumption (#7733) 2021-05-28 19:40:23 +05:30
optimization Avoid adding `None` loss values in `training_epoch_end` (#7772) 2021-05-31 19:28:28 +00:00
properties Refactor PyTorch profiler 4/5 (#6349) 2021-03-23 18:13:29 +01:00
__init__.py
test_config_validator.py Add trainer.predict config validation (#6543) 2021-03-21 21:07:54 +00:00
test_data_loading.py Auto-set `DataLoader.worker_init_fn` with `seed_everything` (#6960) 2021-04-19 16:28:37 +02:00
test_dataloaders.py Add `dataloader_idx` to batch transfer hooks (#6241) 2021-05-13 23:03:55 +05:30
test_progress.py [2/N] Define dataclasses for progress tracking (#7574) 2021-05-22 03:09:08 +02:00
test_states.py `TrainerState` refactor [5/5] (#7173) 2021-05-04 12:50:56 +02:00
test_supporters.py Bugfix/Multiple dataloaders (#7433) 2021-05-11 16:33:29 +02:00
test_trainer.py Remove ProfilerConnector class (#7654) 2021-05-24 08:58:15 -07:00
test_trainer_cli.py add forgotten test in #7240 (#7283) 2021-05-03 23:56:30 +00:00
test_trainer_tricks.py Code cleaning in preparation for #7258 [3/n] (#7262) 2021-04-29 14:40:51 +02:00