chensjtu / GaussianObject

GaussianObject: High-Quality 3D Object Reconstruction from Four Views with Gaussian Splatting (SIGGRAPH Asia 2024, TOG)
887 stars 54 forks source link

train_lora doesn't generate .ckpt file #19

Open ShengL333 opened 7 months ago

ShengL333 commented 7 months ago

I run the train_lora.py using

#LoRA Fine-Tuning stage
python train_lora.py --exp_name controlnet_finetune/kitchen \
    --prompt xxy5syt00 --sh_degree 2 --resolution 8 --sparse_num 4 \
    --data_dir data/mip360/kitchen \
    --gs_dir output/gs_init/kitchen \
    --loo_dir output/gs_init/kitchen_loo \
    --bg_white --sd_locked --train_lora --use_prompt_list \
    --add_diffusion_lora --add_control_lora --add_clip_lora \
    --image_size 256

Then I try to run the train_repair.py using the following

#Gaussian Repair stage
python train_repair.py \
    --config configs/gaussian-object.yaml \
    --train --gpu 0 \
    tag="kitchen" \
    system.init_dreamer="output/gs_init/kitchen" \
    system.exp_name="output/controlnet_finetune/kitchen" \
    system.refresh_size=8 \
    data.data_dir="data/mip360/kitchen" \
    data.resolution=8 \
    data.sparse_num=4 \
    data.prompt="a photo of a xxy5syt00" \
    data.refresh_size=8 \
    system.sh_degree=2 \
    system.novel_image_size=256

The train_repair.py gave me FileNotFoundError: [Errno 2] No such file or directory: 'output/controlnet_finetune/kitchen/ckpts-lora/lora-step=1799.ckpt I think this file is generated by the train_lora step. May someone know the answer to this question?

GaussianObject commented 7 months ago

Thank you for reaching out with your issue! Could you please check if there are any files generated in the output/controlnet_finetune directory after running train_lora.py? Your insights on this will help us assist you more effectively.