MCPcopy Create free account
hub / github.com/THUDM/GLM / __iter__

Method __iter__

tasks/data_utils.py:353–359  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

351 self.num_iters = num_iters
352
353 def __iter__(self):
354 if self.num_iters is not None:
355 for _ in range(self.num_iters):
356 yield None
357 else:
358 while True:
359 yield None
360
361
362def build_data_loader(dataset, batch_size, num_workers, drop_last, shuffle=True, only_rank0=False):

Callers 1

Calls

no outgoing calls

Tested by

no test coverage detected