Pipeline

Basic PyTorch pipeline for general training.

Helper

---

source

CheckpointCB

 CheckpointCB (ck_interval=None, ck_path=None)

Initialize self. See help(type(self)) for accurate signature.

Pipeline

---

source

PipelineIO

 PipelineIO ()

A class providing basic IO functionality.

---

source

Pipeline

 Pipeline (model:torch.nn.modules.module.Module, device:torch.device)

A PipelineIO class providing basic pytorch model training functionality.