src/transformers/models/graphormer/modeling_graphormer.py/LayerDropModuleList/__iter__
class LayerDropModuleList:    def __iter__(self):
        dropout_probs = torch.empty(len(self)).uniform_()
        for i, m in enumerate(super().__iter__()):
            if not self.training or (dropout_probs[i] > self.p):
                yield m