From f5f8e4266f79ef678f472d21f0a923a559beb89c Mon Sep 17 00:00:00 2001 From: hofee Date: Wed, 30 Oct 2024 15:49:11 +0000 Subject: [PATCH] global_only: ratio --- configs/server/server_train_config.yaml | 2 +- core/nbv_dataset.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/configs/server/server_train_config.yaml b/configs/server/server_train_config.yaml index 834ac60..3c848a5 100644 --- a/configs/server/server_train_config.yaml +++ b/configs/server/server_train_config.yaml @@ -13,7 +13,7 @@ runner: epoch: -1 # -1 stands for last epoch max_epochs: 5000 save_checkpoint_interval: 1 - test_first: False + test_first: True train: optimizer: diff --git a/core/nbv_dataset.py b/core/nbv_dataset.py index 4bdb89b..ca9c0c7 100644 --- a/core/nbv_dataset.py +++ b/core/nbv_dataset.py @@ -34,7 +34,7 @@ class NBVReconstructionDataset(BaseDataset): #self.model_dir = config["model_dir"] self.filter_degree = config["filter_degree"] if self.type == namespace.Mode.TRAIN: - scale_ratio = 50 + scale_ratio = 1 self.datalist = self.datalist*scale_ratio if self.cache: expr_root = ConfigManager.get("runner", "experiment", "root_dir")