Package io.github.kirstenali.deepj.persistence

Class TrainingCheckpoint

java.lang.Object

io.github.kirstenali.deepj.persistence.TrainingCheckpoint

public final class TrainingCheckpoint
extends Object

Method Summary

Modifier and Type	Method	Description
static TrainingProgress	load(DecoderOnlyModel model, AdamW optimizer, StatefulBatchSource dataset, CosineLearningRateSchedule schedule, Path path)
static TrainingProgress	load(List<Parameter> params, AdamW optimizer, StatefulBatchSource dataset, CosineLearningRateSchedule schedule, Path path)
static boolean	matches(Path path)
static void	save(DecoderOnlyModel model, AdamW optimizer, StatefulBatchSource dataset, TrainingProgress progress, CosineLearningRateSchedule schedule, Path path)
static void	save(List<Parameter> params, AdamW optimizer, StatefulBatchSource dataset, TrainingProgress progress, CosineLearningRateSchedule schedule, Path path)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Details

matches

public static boolean matches(Path path)
                       throws IOException

Throws:

IOException

save

public static void save(List<Parameter> params,
 AdamW optimizer,
 StatefulBatchSource dataset,
 TrainingProgress progress,
 CosineLearningRateSchedule schedule,
 Path path)
                 throws IOException

Throws:

IOException

save

public static void save(DecoderOnlyModel model,
 AdamW optimizer,
 StatefulBatchSource dataset,
 TrainingProgress progress,
 CosineLearningRateSchedule schedule,
 Path path)
                 throws IOException

Throws:

IOException

load

public static TrainingProgress load(List<Parameter> params,
 AdamW optimizer,
 StatefulBatchSource dataset,
 CosineLearningRateSchedule schedule,
 Path path)
                             throws IOException

Throws:

IOException

load

public static TrainingProgress load(DecoderOnlyModel model,
 AdamW optimizer,
 StatefulBatchSource dataset,
 CosineLearningRateSchedule schedule,
 Path path)
                             throws IOException

Throws:

IOException