[Speedster] 优化模型耗时 10 小时,还没有结束

      def __init__(self, model):
          super().__init__()
          self.model = model
  
      def forward(self, xT, cond2):
          img = self.model.render_speedster(xT, cond2)   
          return img 
          
  xT_model = SmilingXT(model).to(device).eval()
  
  input_data = [((torch.randn(1, 3, 256, 256), torch.randn(1, 512)), torch.tensor([0])) for _ in range(100)]
  
  xT_optimized_model = optimized_model = optimize_model(
    xT_model, input_data=input_data, optimization_time="unconstrained", device=device)   
  
  save_model(xT_optimized_model, "speedster/xT_optimized_model")