Parameter problem about "merge_submodules.py"
ZiyangYan opened this issue · 2 comments
In the last step of training you ask us to input
--config_file configs/mega-nerf/${DATASET_NAME}.yaml
--ckpt_prefix ${EXP_PREFIX}-
--centroid_path ${MASK_PATH}/params.pt
--output $MERGED_O
I just run "Train each submodule" once, and the checkpoints are saved in "/media/zyan/sandiskSSD/datasets/Quad_6k/quad-pixsfm/experiment/10/models"
In the second step, my parameter is
"mega_nerf/train.py --config_file configs/mega-nerf/quad.yaml --exp_name /media/zyan/sandiskSSD/datasets/Quad_6k/quad-pixsfm/experiment --dataset_path /media/zyan/sandiskSSD/datasets/Quad_6k/quad-pixsfm --chunk_paths /media/zyan/sandiskSSD/datasets/Quad_6k/quad-pixsfm/scratch --cluster_mask_path /media/zyan/sandiskSSD/datasets/Quad_6k/quad-pixsfm/mask --train_every 8"
so I think the "--centroid_path" should be --cluster_mask_path/params.pt = media/zyan/sandiskSSD/datasets/Quad_6k/quad-pixsfm/mask/params.pt
But when I open your code, I found you use the "ckpt_prefix" to check whether centroid is exist or not, what it means? Is it still meaningful for us to input centrold_path in the command?
And I always get error about can not find centrold_path like:
Traceback (most recent call last):
File "/media/zyan/sandiskSSD/NeRF_methods/mega-nerf/mega-nerf-main/scripts/merge_submodules.py", line 102, in
main(_get_merge_opts())
File "/media/zyan/sandiskSSD/anaconda3/envs/mega-nerf/lib/python3.9/site-packages/torch/autograd/grad_mode.py", line 28, in decorate_context
return func(*args, **kwargs)
File "/media/zyan/sandiskSSD/NeRF_methods/mega-nerf/mega-nerf-main/scripts/merge_submodules.py", line 38, in main
raise Exception('{} not found'.format(centroid_path))
Exception: /media/zyan/sandiskSSD/datasets/Quad_6k/quad-pixsfm/experiment/100 not found
My input command is:
python scripts/merge_submodules.py --config_file configs/mega-nerf/quad.yaml --ckpt_prefix /media/zyan/sandiskSSD/datasets/Quad_6k/quad-pixsfm/experiment/10 --centroid_path /media/zyan/sandiskSSD/datasets/Quad_6k/quad-pixsfm/mask/params.pt --output /media/zyan/sandiskSSD/datasets/Quad_6k/quad-pixsfm/output
Can you describe more about the parameters for me? thank you
fixed