after first overfit test
This commit is contained in:
@@ -14,8 +14,8 @@ runner:
|
||||
voxel_threshold: 0.01
|
||||
overlap_threshold: 0.5
|
||||
filter_degree: 75
|
||||
to_specified_dir: True # if True, output_dir is used, otherwise, root_dir is used
|
||||
save_points: False
|
||||
to_specified_dir: False # if True, output_dir is used, otherwise, root_dir is used
|
||||
save_points: True
|
||||
save_best_combined_points: True
|
||||
save_mesh: True
|
||||
overwrite: False
|
||||
|
@@ -1,18 +1,18 @@
|
||||
|
||||
runner:
|
||||
general:
|
||||
seed: 0
|
||||
seed: 1
|
||||
device: cuda
|
||||
cuda_visible_devices: "0,1,2,3,4,5,6,7"
|
||||
parallel: False
|
||||
|
||||
experiment:
|
||||
name: test_overfit
|
||||
name: new_test_overfit_2
|
||||
root_dir: "experiments"
|
||||
use_checkpoint: False
|
||||
epoch: -1 # -1 stands for last epoch
|
||||
max_epochs: 5000
|
||||
save_checkpoint_interval: 1
|
||||
save_checkpoint_interval: 3
|
||||
test_first: False
|
||||
|
||||
train:
|
||||
@@ -32,22 +32,29 @@ runner:
|
||||
dataset:
|
||||
OmniObject3d_train:
|
||||
root_dir: "../data/sample_for_training/scenes"
|
||||
model_dir: "../data/scaled_object_meshes"
|
||||
source: nbv_reconstruction_dataset
|
||||
split_file: "../data/sample_for_training/OmniObject3d_train.txt"
|
||||
ratio: 1.0
|
||||
batch_size: 1
|
||||
type: train
|
||||
cache: True
|
||||
ratio: 1
|
||||
batch_size: 128
|
||||
num_workers: 12
|
||||
pts_num: 4096
|
||||
|
||||
OmniObject3d_test:
|
||||
root_dir: "../data/sample_for_training/scenes"
|
||||
model_dir: "../data/scaled_object_meshes"
|
||||
source: nbv_reconstruction_dataset
|
||||
split_file: "../data/sample_for_training/OmniObject3d_train.txt"
|
||||
type: test
|
||||
cache: True
|
||||
filter_degree: 75
|
||||
eval_list:
|
||||
- pose_diff
|
||||
ratio: 0.1
|
||||
batch_size: 1
|
||||
num_workers: 1
|
||||
num_workers: 12
|
||||
pts_num: 4096
|
||||
|
||||
pipeline:
|
||||
@@ -92,4 +99,6 @@ loss_function:
|
||||
gf_loss:
|
||||
|
||||
evaluation_method:
|
||||
pose_diff:
|
||||
pose_diff:
|
||||
coverage_rate_increase:
|
||||
renderer_path: "../blender/data_renderer.py"
|
Reference in New Issue
Block a user