mirror of
https://github.com/deepseek-ai/DualPipe
synced 2025-06-26 18:16:46 +00:00
correct typo
This commit is contained in:
parent
092151c44b
commit
036c14e7f6
@ -52,7 +52,7 @@ class PipelineStage(nn.Module):
|
|||||||
return x
|
return x
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def overlaped_forward_backward(
|
def overlapped_forward_backward(
|
||||||
cls,
|
cls,
|
||||||
module0: "PipelineStage",
|
module0: "PipelineStage",
|
||||||
inputs0: List[torch.Tensor],
|
inputs0: List[torch.Tensor],
|
||||||
|
Loading…
Reference in New Issue
Block a user