diff --git a/configs/server/server_split_dataset_config.yaml b/configs/server/server_split_dataset_config.yaml index 5dd61ac..5a5b97e 100644 --- a/configs/server/server_split_dataset_config.yaml +++ b/configs/server/server_split_dataset_config.yaml @@ -6,17 +6,17 @@ runner: cuda_visible_devices: "0,1,2,3,4,5,6,7" experiment: - name: debug + name: server_split_dataset root_dir: "experiments" split: # - root_dir: "/data/hofee/data/packed_preprocessed_data" + root_dir: "/data/hofee/data/new_full_data" type: "unseen_instance" # "unseen_category" datasets: OmniObject3d_train: - path: "/data/hofee/data/OmniObject3d_train.txt" + path: "/data/hofee/data/new_full_data_list/OmniObject3d_train.txt" ratio: 0.9 OmniObject3d_test: - path: "/data/hofee/data/OmniObject3d_test.txt" + path: "/data/hofee/data/new_full_data_list/OmniObject3d_test.txt" ratio: 0.1 \ No newline at end of file diff --git a/core/pipeline.py b/core/pipeline.py index 1a29fec..e295c87 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(