pytorch_lightning/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

	__init__.py	1.6 KB
	base.py	5.6 KB
	early_stopping.py	7.4 KB
	finetuning.py	13.6 KB
	gpu_stats_monitor.py	8.6 KB
	gradient_accumulation_scheduler.py	2.7 KB
	lambda_function.py	6.9 KB
	lr_monitor.py	7.8 KB
	model_checkpoint.py	24.9 KB
	progress.py	15.1 KB
	pruning.py	15.3 KB