Frosting icon indicating copy to clipboard operation
Frosting copied to clipboard

Errors saving models after optimization

Open gradeeterna opened this issue 1 year ago • 0 comments

Hey,

I'm getting the following errors at the saving model part after optimizing for 15k steps.

Saving model... Traceback (most recent call last): File "H:\Frosting\train.py", line 169, in <module> coarse_sugar_path = coarse_training_with_density_regularization_and_dn_consistency(coarse_args) File "H:\Frosting\frosting_trainers\coarse_density_and_dn_consistency.py", line 902, in coarse_training_with_density_regularization_and_dn_consistency sugar.save_model(path=model_path, File "H:\Frosting\frosting_scene\sugar_model.py", line 2394, in save_model torch.save(checkpoint, path) File "H:\anacondaH\frosting\lib\site-packages\torch\serialization.py", line 440, in save with _open_zipfile_writer(f) as opened_zipfile: File "H:\anacondaH\frosting\lib\site-packages\torch\serialization.py", line 315, in _open_zipfile_writer return container(name_or_buffer) File "H:\anacondaH\frosting\lib\site-packages\torch\serialization.py", line 288, in __init__ super().__init__(torch._C.PyTorchFileWriter(str(name))) RuntimeError: Parent directory ./output/coarse\allotment\sugarcoarse_3Dgs7000_densityestim02_sdfnorm02 does not exist.

I also had to use --gs_output_dir to load an already trained vanilla gs model, as when I try to train from scratch without that argument I get these errors:

Loading config output\vanilla_gs\data/allotment\... Traceback (most recent call last): File "H:\Frosting\train.py", line 169, in <module> coarse_sugar_path = coarse_training_with_density_regularization_and_dn_consistency(coarse_args) File "H:\Frosting\frosting_trainers\coarse_density_and_dn_consistency.py", line 377, in coarse_training_with_density_regularization_and_dn_consistency nerfmodel = GaussianSplattingWrapper( File "H:\Frosting\frosting_scene\gs_model.py", line 132, in __init__ cam_list = load_gs_cameras( File "H:\Frosting\frosting_scene\cameras.py", line 34, in load_gs_cameras with open(gs_output_path + 'cameras.json') as f: FileNotFoundError: [Errno 2] No such file or directory: 'output\\vanilla_gs\\data/allotment\\cameras.json'

Not sure if related, but I'm using pytorch3d 0.7.5 instead of 0.7.4, as "conda install pytorch3d==0.7.4 -c pytorch3d" did not find any packages and I failed to bild it myself, so I used this method instead which only has 0.7.5 up -https://github.com/facebookresearch/pytorch3d/discussions/1752

gradeeterna avatar Sep 25 '24 11:09 gradeeterna