Method encode

diff2flow/ema.py:235–236

(self, *args, **kwargs)

        return self.ema_model.decode(*args, **kwargs)
    
    def encode(self, *args, **kwargs):
        return self.ema_model.encode(*args, **kwargs)
    
    def generate(self, *args, **kwargs):
        return self.ema_model.generate(*args, **kwargs)