Skip to content

GitLab

Unverified Commit 4ac8bfb0 authored by CZYCW's avatar CZYCW Committed by GitHub
Browse files

[NFC] polish colossalai/engine/gradient_handler/utils.py code style (#2708)

parent 6427c406
Show whitespace changes
Inline Side-by-side
Showing with 30 additions and 29 deletions
colossalai/engine/gradient_handler/utils.py
View file @ 4ac8bfb0
from typing import Iterable
import torch.distributed as dist import torch.distributed as dist
import torch.nn as nn import torch.nn as nn
from torch._utils import _flatten_dense_tensors, _unflatten_dense_tensors from torch._utils import _flatten_dense_tensors, _unflatten_dense_tensors
from typing import Iterable
def bucket_allreduce(param_list: Iterable[nn.Parameter], group=None): def bucket_allreduce(param_list: Iterable[nn.Parameter], group=None):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment