From 8a05b7883d6f1568fa397e27e35d6ced87677dd6 Mon Sep 17 00:00:00 2001
From: hofee <lexhofee@gmail.com>
Date: Wed, 30 Oct 2024 15:46:15 +0000
Subject: [PATCH] global_only: train

---
 configs/server/server_train_config.yaml | 22 +++++++++++-----------
 core/pipeline.py                        |  3 ---
 2 files changed, 11 insertions(+), 14 deletions(-)

diff --git a/configs/server/server_train_config.yaml b/configs/server/server_train_config.yaml
index e5451f3..834ac60 100644
--- a/configs/server/server_train_config.yaml
+++ b/configs/server/server_train_config.yaml
@@ -7,7 +7,7 @@ runner:
     parallel: False
     
   experiment:
-    name: overfit_ab_global_only
+    name: train_ab_global_only
     root_dir: "experiments"
     use_checkpoint: False
     epoch: -1 # -1 stands for last epoch
@@ -25,52 +25,52 @@ runner:
   test:
     frequency: 3 # test frequency
     dataset_list:
-      #- OmniObject3d_test
+      - OmniObject3d_test
       - OmniObject3d_val
 
   pipeline: nbv_reconstruction_pipeline
   
 dataset:
   OmniObject3d_train:
-    root_dir: "/data/hofee/nbv_rec_part2_preprocessed"
+    root_dir: "/data/hofee/data/new_full_data"
     model_dir: "../data/scaled_object_meshes"
     source: nbv_reconstruction_dataset
-    split_file: "/data/hofee/data/sample.txt"
+    split_file: "/data/hofee/data/new_full_data_list/OmniObject3d_train.txt"
     type: train
     cache: True
     ratio: 1
     batch_size: 80
-    num_workers: 16
+    num_workers: 128
     pts_num: 8192
     load_from_preprocess: True
 
   OmniObject3d_test:
-    root_dir: "/data/hofee/nbv_rec_part2_preprocessed"
+    root_dir: "/data/hofee/data/new_full_data"
     model_dir: "../data/scaled_object_meshes"
     source: nbv_reconstruction_dataset
-    split_file: "/data/hofee/data/sample.txt"
+    split_file: "/data/hofee/data/new_full_data_list/OmniObject3d_test.txt"
     type: test
     cache: True
     filter_degree: 75
     eval_list:
       - pose_diff
-    ratio: 1
+    ratio: 0.1
     batch_size: 80
     num_workers: 12
     pts_num: 8192
     load_from_preprocess: True
 
   OmniObject3d_val:
-    root_dir: "/data/hofee/nbv_rec_part2_preprocessed"
+    root_dir: "/data/hofee/data/new_full_data"
     model_dir: "../data/scaled_object_meshes"
     source: nbv_reconstruction_dataset
-    split_file: "/data/hofee/data/sample.txt"
+    split_file: "/data/hofee/data/new_full_data_list/OmniObject3d_train.txt"
     type: test
     cache: True
     filter_degree: 75
     eval_list:
       - pose_diff
-    ratio: 1
+    ratio: 0.01
     batch_size: 80
     num_workers: 12
     pts_num: 8192
diff --git a/core/pipeline.py b/core/pipeline.py
index 0fad455..a43d572 100644
--- 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(