MCPcopy Create free account
hub / github.com/AIS-SNU/Smart-Infinity / set_epoch

Method set_epoch

DeepSpeedExample/megatron/data/samplers.py:74–75  ·  view source on GitHub ↗
(self, epoch)

Source from the content-addressed store, hash-verified

72 return self.num_samples
73
74 def set_epoch(self, epoch):
75 self.epoch = epoch
76
77
78class DistributedBatchSampler(data.sampler.BatchSampler):

Callers 1

_trainFunction · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected