mirror of https://github.com/InternLM/InternLM
style(solver/optimizer/utils.py): fix lint error (#147)
Co-authored-by: huangting.p <huangting@sensetime.com>pull/158/head
parent
fcc3534509
commit
8b1717a05d
|
@ -1,11 +1,11 @@
|
||||||
#!/usr/bin/env python
|
#!/usr/bin/env python
|
||||||
# -*- encoding: utf-8 -*-
|
# -*- encoding: utf-8 -*-
|
||||||
|
|
||||||
|
import math
|
||||||
from abc import ABC, abstractmethod
|
from abc import ABC, abstractmethod
|
||||||
from typing import Dict, Optional
|
from typing import Dict, Optional
|
||||||
|
|
||||||
import amp_C
|
import amp_C
|
||||||
import math
|
|
||||||
import torch
|
import torch
|
||||||
import torch.distributed as dist
|
import torch.distributed as dist
|
||||||
from apex.multi_tensor_apply import multi_tensor_applier
|
from apex.multi_tensor_apply import multi_tensor_applier
|
||||||
|
|
Loading…
Reference in New Issue