Skip to content

GitLab

Commit 1559a09f authored by zbian's avatar zbian Committed by binmakeswell
Browse files

[NFC] polish amp.naive_amp.grad_scaler code style

parent e9635eb4
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 3 deletions
colossalai/amp/naive_amp/grad_scaler/base_grad_scaler.py
View file @ 1559a09f
#!/usr/bin/env python
# -*- encoding: utf-8 -*-
import torch
from abc import ABC, abstractmethod
from colossalai.logging import get_dist_logger
from torch import Tensor
from typing import Dict
import torch
from torch import Tensor
from colossalai.logging import get_dist_logger
__all__ = ['BaseGradScaler']
......
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