github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-20 18:30:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

[NFC] fix format of application/Chat/coati/trainer/utils.py (#4273)

Browse Source
This commit is contained in:
アマデウス 2023-07-18 18:02:35 +08:00 committed by binmakeswell
parent 1ce997daaf
commit caa4433072
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
applications/Chat/coati/trainer/utils.py
View File

@ -14,9 +14,10 @@ class CycledDataLoader:
NOTE: next(iter(dataloader)) is not equivalent to for batch in dataloader: break, it causes slightly different behavior. NOTE: next(iter(dataloader)) is not equivalent to for batch in dataloader: break, it causes slightly different behavior.
""" """
def __init__(self, def __init__(
dataloader: DataLoader, self,
) -> None: dataloader: DataLoader,
) -> None:
self.dataloader = dataloader self.dataloader = dataloader
self.count = 0 self.count = 0