Skip to content

1zb/GeomDist

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

35 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Geometry Distributions

🚅 Training

torchrun --nproc_per_node=4 main.py --blr 5e-7 --output_dir output/loong --log_dir output/loong --data_path shapes/loong.obj

🎈 Inference

python infer.py --pth output/loong/checkpoint-999.pth --target Gaussian --num-steps 64 --output samples/loong --N 10000000

💾 Datasets

https://huggingface.co/datasets/Zbalpha/shapes

💼 Checkpoints

https://huggingface.co/Zbalpha/geom_dist_ckpt

📧 Contact

Contact Biao Zhang (@1zb) if you have any further questions. This repository is for academic research use only.

📘 Citation

@article{zhang2024geometry,
  title={Geometry Distributions},
  author={Zhang, Biao and Ren, Jing and Wonka, Peter},
  journal={arXiv preprint arXiv:2411.16076},
  year={2024}
}

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages