Method parameter_rrefs

distributed/rpc/pipeline/main.py:77–82

r""" Create one RRef for each parameter in the given local module, and return a list of RRefs.

(self)

        return nn.Sequential(*layers)

    def parameter_rrefs(self):
        r"""
        Create one RRef for each parameter in the given local module, and return a
        list of RRefs.
        """
        return [RRef(p) for p in self.parameters()]


class ResNetShard1(ResNetBase):