Uses of Record Class
io.github.kirstenali.deepj.training.CosineLearningRateSchedule
Packages that use CosineLearningRateSchedule
-
Uses of CosineLearningRateSchedule in io.github.kirstenali.deepj.persistence
Methods in io.github.kirstenali.deepj.persistence with parameters of type CosineLearningRateScheduleModifier and TypeMethodDescriptionstatic TrainingProgressTrainingCheckpoint.load(DecoderOnlyModel model, AdamW optimizer, StatefulBatchSource dataset, CosineLearningRateSchedule schedule, Path path) static TrainingProgressTrainingCheckpoint.load(List<Parameter> params, AdamW optimizer, StatefulBatchSource dataset, CosineLearningRateSchedule schedule, Path path) static voidTrainingCheckpoint.save(DecoderOnlyModel model, AdamW optimizer, StatefulBatchSource dataset, TrainingProgress progress, CosineLearningRateSchedule schedule, Path path) static voidTrainingCheckpoint.save(List<Parameter> params, AdamW optimizer, StatefulBatchSource dataset, TrainingProgress progress, CosineLearningRateSchedule schedule, Path path)