def step()

in dualpipe/dualpipev.py [0:0]

• 7 lines of code
• 1 McCabe index (conditional complexity)

    def step(
        self,
        *inputs: Optional[torch.Tensor],
        num_chunks: int = 0,
        criterion: Optional[Callable] = None,
        labels: List[Optional[torch.Tensor]] = [],
        return_outputs: bool = False,