Method to

src/modules/pixie/pixie_encoder.py:134–138

(self, device)

                net.eval()
    
    def to(self, device):
        for module in [self.Encoder, self.Regressor, self.Extractor, self.Moderator]:
            for net in module.values():
                net = net.to(device)
        self.smplx.to(device)
    
    def eval(self):
        for module in [self.Encoder, self.Regressor, self.Extractor, self.Moderator]: