mirror of https://github.com/hpcaitech/ColossalAI
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.cipull/6001/head
parent
10b72a32b1
commit
b0c89bf298
|
@ -9,8 +9,6 @@ import torch.distributed as dist
|
|||
from torch.utils._pytree import tree_map
|
||||
from torch.utils.data import DataLoader
|
||||
|
||||
from colossalai.booster import Booster
|
||||
|
||||
|
||||
class CycledDataLoader:
|
||||
"""
|
||||
|
|
Loading…
Reference in New Issue