larry._callbacks

Submodules

Package Contents

Classes

InterpolationTask

class larry._callbacks.InterpolationTask(adata: anndata.AnnData, potential: bool = False, time_key: str = 'Time point', lineage_key: str = 'clone_idx', use_key: str = 'X_pca', seed: int = 617, silent: bool = True, device=AutoDevice())

Bases: pytorch_lightning.Callback

configure_eval_data()
forward_without_grad(DiffEq)

Forward integrate over the model without gradients.

forward_with_grad(DiffEq)

Forward integrate over the model retaining gradients.

on_train_epoch_end(trainer, DiffEq)

Call