Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
lightning/tests/tests_pytorch/trainer
History
Carlos Mocholí b97b3ac053
Avoid `inference_mode` with `torch.compile` (#17215) 		2023-03-29 19:43:28 +00:00
..
connectors Set a fixed stage in the evaluation loops (#17094) 2023-03-27 10:00:03 +02:00
dynamic_args Remove memory-retaining epoch-end hooks (#16520) 2023-02-06 17:00:36 +00:00
flags Avoid `inference_mode` with `torch.compile` (#17215) 2023-03-29 19:43:28 +00:00
logging_ Typo fix: on_training_epoch_end -> on_train_epoch_end (#17110) 2023-03-17 10:01:00 +00:00
optimization Remove the `*_step_end` hooks (#16791) 2023-02-20 13:04:40 +00:00
properties Set a fixed stage in the evaluation loops (#17094) 2023-03-27 10:00:03 +02:00
__init__.py
…
test_config_validator.py Remove DP (#16748) 2023-02-16 05:12:08 +00:00
test_dataloaders.py Set a fixed stage in the evaluation loops (#17094) 2023-03-27 10:00:03 +02:00
test_states.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
test_trainer.py Set a fixed stage in the evaluation loops (#17094) 2023-03-27 10:00:03 +02:00