global_only: train

2024-10-30 15:46:15 +00:00
parent e23697eb87
commit 8a05b7883d
2 changed files with 11 additions and 14 deletions
--- a/core/pipeline.py
+++ b/core/pipeline.py
@@ -54,10 +54,7 @@ class NBVReconstructionPipeline(nn.Module):
        return perturbed_x, random_t, target_score, std

    def forward_train(self, data):
-        start_time = time.time()
        main_feat = self.get_main_feat(data)
-        end_time = time.time()
-        print("get_main_feat time: ", end_time - start_time)
        """ get std """
        best_to_world_pose_9d_batch = data["best_to_world_pose_9d"]
        perturbed_x, random_t, target_score, std = self.pertube_data(