Pipeline
Basic PyTorch pipeline for general training.
Helper
CheckpointCB
def CheckpointCB(
ck_interval:NoneType=None, ck_path:NoneType=None
):
Initialize self. See help(type(self)) for accurate signature.
Pipeline
PipelineIO
def PipelineIO(
args:VAR_POSITIONAL, kwargs:VAR_KEYWORD
):
A class providing basic IO functionality.
Pipeline
def Pipeline(
model:Module, device:device
):
A PipelineIO class providing basic pytorch model training functionality.