Update apps/infer.py

apps/infer.py

@@ -450,8 +450,6 @@ def generate_model(in_path, model_type):
     video_path = os.path.join(config_dict['out_dir'], cfg.name, f"vid/{data['name']}_cloth.mp4")
     overlap_path = os.path.join(config_dict['out_dir'], cfg.name, f"png/{data['name']}_overlap.png")
     
-    torch.cuda.empty_cache()
-    
     del model
     del dataset
     del local_affine_model
@@ -462,4 +460,6 @@ def generate_model(in_path, model_type):
     del losses
     del in_tensor
     
+    torch.cuda.empty_cache()
+    
     return [smpl_path, smpl_path, smpl_npy_path, recon_path, recon_path, refine_path, refine_path, video_path, overlap_path]