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

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

0 0 0 Python

	progress
	__init__.py
	test_callback_hook_outputs.py	1.8 KB
	test_callbacks.py	5.9 KB
	test_device_stats_monitor.py	7.5 KB
	test_early_stopping.py	25.0 KB
	test_finetuning_callback.py	16.3 KB
	test_gradient_accumulation_scheduler.py	5.6 KB
	test_lambda_function.py	2.1 KB
	test_lr_monitor.py	29.1 KB
	test_model_checkpoint_additional_cases.py	7.4 KB
	test_model_checkpoint_edge_cases.py	5.8 KB
	test_model_checkpoint_manual_opt.py	6.4 KB
	test_model_checkpoint_step_interval_val_metric.py	3.5 KB
	test_model_summary.py	2.8 KB
	test_prediction_writer.py	5.3 KB
	test_pruning.py	21.5 KB
	test_rich_model_summary.py	3.5 KB
	test_spike.py	4.7 KB
	test_stochastic_weight_avg.py	16.1 KB
	test_throughput_monitor.py	17.3 KB
	test_timer.py	7.3 KB
	test_weight_averaging.py	17.0 KB