Skip to content

GitLab

Commit b25030cc authored by Genghan Zhang's avatar Genghan Zhang Committed by binmakeswell
Browse files

[NFC] polish ./colossalai/amp/torch_amp/__init__.py code style (#1836)

parent b0a138aa
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 3 deletions
colossalai/amp/torch_amp/__init__.py
View file @ b25030cc
from typing import Optional
import torch.nn as nn
from torch.optim import Optimizer
from torch.nn.modules.loss import _Loss
from torch.optim import Optimizer
from colossalai.context import Config
from .torch_amp import TorchAMPOptimizer, TorchAMPModel, TorchAMPLoss
from typing import Optional
from .torch_amp import TorchAMPLoss, TorchAMPModel, TorchAMPOptimizer
def convert_to_torch_amp(model: nn.Module,
......
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