reconstructing ref mesh

human2b · May 25, 2022 · 46c760b · 46c760b
1 parent afe5526
commit 46c760b
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/model/fit_i3DMM_to_mesh.py b/model/fit_i3DMM_to_mesh.py
@@ -314,12 +314,12 @@ def empirical_stat(latent_vecs, indices):
             decoder_col.load_state_dict(saved_model_state_col["model_state_dict"])
         else:
             saved_model_epoch = saved_model_state_ref["epoch"]
-            saved_model_state_col = torch.load(
-                os.path.join(
-                    exp_directory, ws.model_params_subdir, args.checkpoint + "_col.pth"
-                )
-            )
-            decoder_col.load_state_dict(saved_model_state_col["model_state_dict"])
+        #    saved_model_state_col = torch.load(
+        #        os.path.join(
+        #            exp_directory, ws.model_params_subdir, args.checkpoint + "_col.pth"
+        #        )
+        #    )
+        #    decoder_col.load_state_dict(saved_model_state_col["model_state_dict"])
 
 
         decoder_deform = decoder_deform.module.cuda()