d192ef06119f3eccf569a844b30976db99cfa0f6,python/ray/util/sgd/torch/distributed_torch_runner.py,DistributedTorchRunner,train_epoch,#DistributedTorchRunner#,106

Before Change

        Automatically sets epoch of sampler if possible.
        
        if hasattr(self.train_loader.sampler, "set_epoch"):
            self.train_loader.sampler.set_epoch(self.epochs)
        return super(DistributedTorchRunner, self).train_epoch(**kwargs)

After Change

        Automatically sets epoch of sampler if possible.
        
        if hasattr(self.train_loader, "sampler") and hasattr(
                self.train_loader.sampler, "set_epoch"):
            self.train_loader.sampler.set_epoch(self.epochs)
        return super(DistributedTorchRunner, self).train_epoch(**kwargs)

Instances