lightning/tests/tests_pytorch/trainer
Rohit Gupta ce702fd40e
Squeeze tensor while logging (#14489)
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
2022-09-05 14:01:51 +00:00
..
connectors Remove deprecated HPC model hooks (#14315) 2022-08-26 20:59:32 +00:00
dynamic_args
flags Fix mypy errors attributed to `pytorch_lightning.trainer.connectors.data_connector.py` (#13806) 2022-08-26 13:28:27 +00:00
logging_ Squeeze tensor while logging (#14489) 2022-09-05 14:01:51 +00:00
optimization Deprecate `amp_level` from `Trainer` (#13898) 2022-08-05 08:31:19 +00:00
properties Estimate stepping batches with max_steps if max_epochs is not set (#14317) 2022-09-05 09:05:21 +00:00
__init__.py
test_config_validator.py Mark stage argument in hooks as required (#14064) 2022-09-01 15:47:40 +02:00
test_dataloaders.py Fix an issue to avoid the impact of sanity check on `reload_dataloaders_every_n_epochs` for validation (#13964) 2022-08-21 23:55:03 +05:30
test_progress.py
test_states.py
test_supporters.py Add support for DDP fork (#13405) 2022-07-22 16:05:35 +00:00
test_trainer.py Change `trainer.should_stop` to not stop in between an epoch and run until `min_steps/min_epochs` only (#13890) 2022-08-27 12:12:24 +00:00
test_trainer_cli.py Add support for DDP fork (#13405) 2022-07-22 16:05:35 +00:00