global: debug inference

This commit is contained in:
2024-11-01 15:47:11 +00:00
parent 985a08d89c
commit ecd4cfa806
4 changed files with 113 additions and 43 deletions

View File

@@ -1,12 +1,12 @@
runner:
general:
seed: 1
seed: 0
device: cuda
cuda_visible_devices: "0,1,2,3,4,5,6,7"
experiment:
name: overfit_ab_global_only
name: train_ab_global_only
root_dir: "experiments"
epoch: -1 # -1 stands for last epoch
@@ -22,7 +22,7 @@ runner:
dataset:
OmniObject3d_train:
root_dir: "/data/hofee/data/new_full_data"
model_dir: "/data/hofee/data/scaled_object_meshes"
model_dir: "/data/hofee/data/object_meshes_part3"
source: seq_reconstruction_dataset
split_file: "/data/hofee/data/sample.txt"
type: test
@@ -35,7 +35,7 @@ dataset:
OmniObject3d_test:
root_dir: "/data/hofee/data/new_full_data"
model_dir: "/data/hofee/data/scaled_object_meshes"
model_dir: "/data/hofee/data/object_meshes_part3"
source: seq_reconstruction_dataset
split_file: "/data/hofee/data/sample.txt"
type: test

View File

@@ -9,7 +9,7 @@ runner:
experiment:
name: train_ab_global_only
root_dir: "experiments"
use_checkpoint: False
use_checkpoint: True
epoch: -1 # -1 stands for last epoch
max_epochs: 5000
save_checkpoint_interval: 1