mirror of https://github.com/hpcaitech/ColossalAI
[NFC] fix format of application/Chat/coati/trainer/utils.py (#4273)
parent
1ce997daaf
commit
caa4433072
|
@ -14,7 +14,8 @@ class CycledDataLoader:
|
|||
NOTE: next(iter(dataloader)) is not equivalent to for batch in dataloader: break, it causes slightly different behavior.
|
||||
"""
|
||||
|
||||
def __init__(self,
|
||||
def __init__(
|
||||
self,
|
||||
dataloader: DataLoader,
|
||||
) -> None:
|
||||
self.dataloader = dataloader
|
||||
|
|
Loading…
Reference in New Issue