tests/tests_pytorch/loops - Lightning-AI/pytorch-lightning

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

30975 0 0 Python

	optimization
	__init__.py
	test_all.py	3.9 KB
	test_double_iter_in_iterable_dataset.py	2.6 KB
	test_evaluation_loop_flow.py	3.6 KB
	test_evaluation_loop.py	18.1 KB
	test_fetchers.py	27.3 KB
	test_flow_warnings.py	1.3 KB
	test_loop_state_dict.py	4.2 KB
	test_loops.py	47.8 KB
	test_prediction_loop.py	10.9 KB
	test_progress.py	3.6 KB
	test_training_epoch_loop.py	8.6 KB
	test_training_loop_flow_dict.py	4.6 KB
	test_training_loop_flow_scalar.py	7.7 KB
	test_training_loop.py	12.1 KB