github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-21 17:40:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

[NFC] polish colossalai/utils/multi_tensor_apply/multi_tensor_apply.py code style (#1559)

Browse Source
This commit is contained in:
Kirigaya Kazuto
2022-09-08 22:04:34 +08:00
committed by GitHub
parent b0f4c0bddf
commit 318fbf1145
2 changed files with 3 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/pipeline/rpc/PipelineBase.py
View File

@@ -778,4 +778,4 @@ class OneFOneBPipelineEngine(PipelineEngineBase):
criterion: Callable = None,
checkpoint: bool = False) -> None:
use_1F1B = True
super().__init__(module_partitions, stage_num, num_microbatches, device, use_1F1B, chunk, criterion, checkpoint)
super().__init__(module_partitions, stage_num, num_microbatches, device, use_1F1B, chunk, criterion, checkpoint)