diff --git a/.gitattributes b/.gitattributes
index a07429ce8adf06f4c157abd7ebd3b955bb29685c..7773be64c2465708ef982364af0a254a95b04e69 100644
--- a/.gitattributes
+++ b/.gitattributes
@@ -33,6 +33,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
-data/datasets/moleculenet/qm9/qm9.csv filter=lfs diff=lfs merge=lfs -text
-data/datasets/moleculenet/qm9/train.csv filter=lfs diff=lfs merge=lfs -text
-images/3dgridvqgan_architecture.png filter=lfs diff=lfs merge=lfs -text
+3dgrid_vqgan/data/datasets/moleculenet/qm9/qm9.csv filter=lfs diff=lfs merge=lfs -text
+3dgrid_vqgan/data/datasets/moleculenet/qm9/train.csv filter=lfs diff=lfs merge=lfs -text
+3dgrid_vqgan/images/3dgridvqgan_architecture.png filter=lfs diff=lfs merge=lfs -text
diff --git a/data/3d_grids_sample/00b781f5a45a2cd1cdff1a582f5650f9.smi b/3dgrid_vqgan/data/3d_grids_sample/00b781f5a45a2cd1cdff1a582f5650f9.smi
similarity index 100%
rename from data/3d_grids_sample/00b781f5a45a2cd1cdff1a582f5650f9.smi
rename to 3dgrid_vqgan/data/3d_grids_sample/00b781f5a45a2cd1cdff1a582f5650f9.smi
diff --git a/data/3d_grids_sample/00b781f5a45a2cd1cdff1a582f5650f9_0.npy b/3dgrid_vqgan/data/3d_grids_sample/00b781f5a45a2cd1cdff1a582f5650f9_0.npy
similarity index 100%
rename from data/3d_grids_sample/00b781f5a45a2cd1cdff1a582f5650f9_0.npy
rename to 3dgrid_vqgan/data/3d_grids_sample/00b781f5a45a2cd1cdff1a582f5650f9_0.npy
diff --git a/data/3d_grids_sample/00b781f5a45a2cd1cdff1a582f5650f9_0.xyz b/3dgrid_vqgan/data/3d_grids_sample/00b781f5a45a2cd1cdff1a582f5650f9_0.xyz
similarity index 100%
rename from data/3d_grids_sample/00b781f5a45a2cd1cdff1a582f5650f9_0.xyz
rename to 3dgrid_vqgan/data/3d_grids_sample/00b781f5a45a2cd1cdff1a582f5650f9_0.xyz
diff --git a/data/3d_grids_sample/0a18e0f64cbaf1508b32834ece70933c.smi b/3dgrid_vqgan/data/3d_grids_sample/0a18e0f64cbaf1508b32834ece70933c.smi
similarity index 100%
rename from data/3d_grids_sample/0a18e0f64cbaf1508b32834ece70933c.smi
rename to 3dgrid_vqgan/data/3d_grids_sample/0a18e0f64cbaf1508b32834ece70933c.smi
diff --git a/data/3d_grids_sample/0a18e0f64cbaf1508b32834ece70933c_0.npy b/3dgrid_vqgan/data/3d_grids_sample/0a18e0f64cbaf1508b32834ece70933c_0.npy
similarity index 100%
rename from data/3d_grids_sample/0a18e0f64cbaf1508b32834ece70933c_0.npy
rename to 3dgrid_vqgan/data/3d_grids_sample/0a18e0f64cbaf1508b32834ece70933c_0.npy
diff --git a/data/3d_grids_sample/0a18e0f64cbaf1508b32834ece70933c_0.xyz b/3dgrid_vqgan/data/3d_grids_sample/0a18e0f64cbaf1508b32834ece70933c_0.xyz
similarity index 100%
rename from data/3d_grids_sample/0a18e0f64cbaf1508b32834ece70933c_0.xyz
rename to 3dgrid_vqgan/data/3d_grids_sample/0a18e0f64cbaf1508b32834ece70933c_0.xyz
diff --git a/data/3d_grids_sample/0aa6c786f42a0d43d8d1dfc7e9ae4939.smi b/3dgrid_vqgan/data/3d_grids_sample/0aa6c786f42a0d43d8d1dfc7e9ae4939.smi
similarity index 100%
rename from data/3d_grids_sample/0aa6c786f42a0d43d8d1dfc7e9ae4939.smi
rename to 3dgrid_vqgan/data/3d_grids_sample/0aa6c786f42a0d43d8d1dfc7e9ae4939.smi
diff --git a/data/3d_grids_sample/0aa6c786f42a0d43d8d1dfc7e9ae4939_0.npy b/3dgrid_vqgan/data/3d_grids_sample/0aa6c786f42a0d43d8d1dfc7e9ae4939_0.npy
similarity index 100%
rename from data/3d_grids_sample/0aa6c786f42a0d43d8d1dfc7e9ae4939_0.npy
rename to 3dgrid_vqgan/data/3d_grids_sample/0aa6c786f42a0d43d8d1dfc7e9ae4939_0.npy
diff --git a/data/3d_grids_sample/0aa6c786f42a0d43d8d1dfc7e9ae4939_0.xyz b/3dgrid_vqgan/data/3d_grids_sample/0aa6c786f42a0d43d8d1dfc7e9ae4939_0.xyz
similarity index 100%
rename from data/3d_grids_sample/0aa6c786f42a0d43d8d1dfc7e9ae4939_0.xyz
rename to 3dgrid_vqgan/data/3d_grids_sample/0aa6c786f42a0d43d8d1dfc7e9ae4939_0.xyz
diff --git a/data/3d_grids_sample/0ae6f92c43122633a151eff5089e8da6.smi b/3dgrid_vqgan/data/3d_grids_sample/0ae6f92c43122633a151eff5089e8da6.smi
similarity index 100%
rename from data/3d_grids_sample/0ae6f92c43122633a151eff5089e8da6.smi
rename to 3dgrid_vqgan/data/3d_grids_sample/0ae6f92c43122633a151eff5089e8da6.smi
diff --git a/data/3d_grids_sample/0ae6f92c43122633a151eff5089e8da6_0.npy b/3dgrid_vqgan/data/3d_grids_sample/0ae6f92c43122633a151eff5089e8da6_0.npy
similarity index 100%
rename from data/3d_grids_sample/0ae6f92c43122633a151eff5089e8da6_0.npy
rename to 3dgrid_vqgan/data/3d_grids_sample/0ae6f92c43122633a151eff5089e8da6_0.npy
diff --git a/data/3d_grids_sample/0ae6f92c43122633a151eff5089e8da6_0.xyz b/3dgrid_vqgan/data/3d_grids_sample/0ae6f92c43122633a151eff5089e8da6_0.xyz
similarity index 100%
rename from data/3d_grids_sample/0ae6f92c43122633a151eff5089e8da6_0.xyz
rename to 3dgrid_vqgan/data/3d_grids_sample/0ae6f92c43122633a151eff5089e8da6_0.xyz
diff --git a/data/3d_grids_sample/0b4020095a14325f0f174bc8a43f625d.smi b/3dgrid_vqgan/data/3d_grids_sample/0b4020095a14325f0f174bc8a43f625d.smi
similarity index 100%
rename from data/3d_grids_sample/0b4020095a14325f0f174bc8a43f625d.smi
rename to 3dgrid_vqgan/data/3d_grids_sample/0b4020095a14325f0f174bc8a43f625d.smi
diff --git a/data/3d_grids_sample/0b4020095a14325f0f174bc8a43f625d_0.npy b/3dgrid_vqgan/data/3d_grids_sample/0b4020095a14325f0f174bc8a43f625d_0.npy
similarity index 100%
rename from data/3d_grids_sample/0b4020095a14325f0f174bc8a43f625d_0.npy
rename to 3dgrid_vqgan/data/3d_grids_sample/0b4020095a14325f0f174bc8a43f625d_0.npy
diff --git a/data/3d_grids_sample/0b4020095a14325f0f174bc8a43f625d_0.xyz b/3dgrid_vqgan/data/3d_grids_sample/0b4020095a14325f0f174bc8a43f625d_0.xyz
similarity index 100%
rename from data/3d_grids_sample/0b4020095a14325f0f174bc8a43f625d_0.xyz
rename to 3dgrid_vqgan/data/3d_grids_sample/0b4020095a14325f0f174bc8a43f625d_0.xyz
diff --git a/data/3d_grids_sample/0b77c16e04a8ac8f84ae9ccaf9b1aaa0.smi b/3dgrid_vqgan/data/3d_grids_sample/0b77c16e04a8ac8f84ae9ccaf9b1aaa0.smi
similarity index 100%
rename from data/3d_grids_sample/0b77c16e04a8ac8f84ae9ccaf9b1aaa0.smi
rename to 3dgrid_vqgan/data/3d_grids_sample/0b77c16e04a8ac8f84ae9ccaf9b1aaa0.smi
diff --git a/data/3d_grids_sample/0b77c16e04a8ac8f84ae9ccaf9b1aaa0_0.npy b/3dgrid_vqgan/data/3d_grids_sample/0b77c16e04a8ac8f84ae9ccaf9b1aaa0_0.npy
similarity index 100%
rename from data/3d_grids_sample/0b77c16e04a8ac8f84ae9ccaf9b1aaa0_0.npy
rename to 3dgrid_vqgan/data/3d_grids_sample/0b77c16e04a8ac8f84ae9ccaf9b1aaa0_0.npy
diff --git a/data/3d_grids_sample/0b77c16e04a8ac8f84ae9ccaf9b1aaa0_0.xyz b/3dgrid_vqgan/data/3d_grids_sample/0b77c16e04a8ac8f84ae9ccaf9b1aaa0_0.xyz
similarity index 100%
rename from data/3d_grids_sample/0b77c16e04a8ac8f84ae9ccaf9b1aaa0_0.xyz
rename to 3dgrid_vqgan/data/3d_grids_sample/0b77c16e04a8ac8f84ae9ccaf9b1aaa0_0.xyz
diff --git a/data/3d_grids_sample/0b88d98ac218831353fb8c61aea0cfe8.smi b/3dgrid_vqgan/data/3d_grids_sample/0b88d98ac218831353fb8c61aea0cfe8.smi
similarity index 100%
rename from data/3d_grids_sample/0b88d98ac218831353fb8c61aea0cfe8.smi
rename to 3dgrid_vqgan/data/3d_grids_sample/0b88d98ac218831353fb8c61aea0cfe8.smi
diff --git a/data/3d_grids_sample/0b88d98ac218831353fb8c61aea0cfe8_0.npy b/3dgrid_vqgan/data/3d_grids_sample/0b88d98ac218831353fb8c61aea0cfe8_0.npy
similarity index 100%
rename from data/3d_grids_sample/0b88d98ac218831353fb8c61aea0cfe8_0.npy
rename to 3dgrid_vqgan/data/3d_grids_sample/0b88d98ac218831353fb8c61aea0cfe8_0.npy
diff --git a/data/3d_grids_sample/0b88d98ac218831353fb8c61aea0cfe8_0.xyz b/3dgrid_vqgan/data/3d_grids_sample/0b88d98ac218831353fb8c61aea0cfe8_0.xyz
similarity index 100%
rename from data/3d_grids_sample/0b88d98ac218831353fb8c61aea0cfe8_0.xyz
rename to 3dgrid_vqgan/data/3d_grids_sample/0b88d98ac218831353fb8c61aea0cfe8_0.xyz
diff --git a/data/3d_grids_sample/0b94b07e1ec5e58964bfc7e670a359fb.smi b/3dgrid_vqgan/data/3d_grids_sample/0b94b07e1ec5e58964bfc7e670a359fb.smi
similarity index 100%
rename from data/3d_grids_sample/0b94b07e1ec5e58964bfc7e670a359fb.smi
rename to 3dgrid_vqgan/data/3d_grids_sample/0b94b07e1ec5e58964bfc7e670a359fb.smi
diff --git a/data/3d_grids_sample/0b94b07e1ec5e58964bfc7e670a359fb_0.npy b/3dgrid_vqgan/data/3d_grids_sample/0b94b07e1ec5e58964bfc7e670a359fb_0.npy
similarity index 100%
rename from data/3d_grids_sample/0b94b07e1ec5e58964bfc7e670a359fb_0.npy
rename to 3dgrid_vqgan/data/3d_grids_sample/0b94b07e1ec5e58964bfc7e670a359fb_0.npy
diff --git a/data/3d_grids_sample/0b94b07e1ec5e58964bfc7e670a359fb_0.xyz b/3dgrid_vqgan/data/3d_grids_sample/0b94b07e1ec5e58964bfc7e670a359fb_0.xyz
similarity index 100%
rename from data/3d_grids_sample/0b94b07e1ec5e58964bfc7e670a359fb_0.xyz
rename to 3dgrid_vqgan/data/3d_grids_sample/0b94b07e1ec5e58964bfc7e670a359fb_0.xyz
diff --git a/data/datasets/moleculenet/qm9/qm9.csv b/3dgrid_vqgan/data/datasets/moleculenet/qm9/qm9.csv
similarity index 100%
rename from data/datasets/moleculenet/qm9/qm9.csv
rename to 3dgrid_vqgan/data/datasets/moleculenet/qm9/qm9.csv
diff --git a/data/datasets/moleculenet/qm9/test.csv b/3dgrid_vqgan/data/datasets/moleculenet/qm9/test.csv
similarity index 100%
rename from data/datasets/moleculenet/qm9/test.csv
rename to 3dgrid_vqgan/data/datasets/moleculenet/qm9/test.csv
diff --git a/data/datasets/moleculenet/qm9/train.csv b/3dgrid_vqgan/data/datasets/moleculenet/qm9/train.csv
similarity index 100%
rename from data/datasets/moleculenet/qm9/train.csv
rename to 3dgrid_vqgan/data/datasets/moleculenet/qm9/train.csv
diff --git a/data/datasets/moleculenet/qm9/valid.csv b/3dgrid_vqgan/data/datasets/moleculenet/qm9/valid.csv
similarity index 100%
rename from data/datasets/moleculenet/qm9/valid.csv
rename to 3dgrid_vqgan/data/datasets/moleculenet/qm9/valid.csv
diff --git a/finetune/args.py b/3dgrid_vqgan/finetune/args.py
similarity index 100%
rename from finetune/args.py
rename to 3dgrid_vqgan/finetune/args.py
diff --git a/finetune/dataset/__init__.py b/3dgrid_vqgan/finetune/dataset/__init__.py
similarity index 100%
rename from finetune/dataset/__init__.py
rename to 3dgrid_vqgan/finetune/dataset/__init__.py
diff --git a/finetune/dataset/default.py b/3dgrid_vqgan/finetune/dataset/default.py
similarity index 100%
rename from finetune/dataset/default.py
rename to 3dgrid_vqgan/finetune/dataset/default.py
diff --git a/finetune/finetune_regression.py b/3dgrid_vqgan/finetune/finetune_regression.py
similarity index 100%
rename from finetune/finetune_regression.py
rename to 3dgrid_vqgan/finetune/finetune_regression.py
diff --git a/finetune/run_finetune_qm9_alpha.sh b/3dgrid_vqgan/finetune/run_finetune_qm9_alpha.sh
similarity index 100%
rename from finetune/run_finetune_qm9_alpha.sh
rename to 3dgrid_vqgan/finetune/run_finetune_qm9_alpha.sh
diff --git a/finetune/run_finetune_qm9_cv.sh b/3dgrid_vqgan/finetune/run_finetune_qm9_cv.sh
similarity index 100%
rename from finetune/run_finetune_qm9_cv.sh
rename to 3dgrid_vqgan/finetune/run_finetune_qm9_cv.sh
diff --git a/finetune/run_finetune_qm9_g298.sh b/3dgrid_vqgan/finetune/run_finetune_qm9_g298.sh
similarity index 100%
rename from finetune/run_finetune_qm9_g298.sh
rename to 3dgrid_vqgan/finetune/run_finetune_qm9_g298.sh
diff --git a/finetune/run_finetune_qm9_gap.sh b/3dgrid_vqgan/finetune/run_finetune_qm9_gap.sh
similarity index 100%
rename from finetune/run_finetune_qm9_gap.sh
rename to 3dgrid_vqgan/finetune/run_finetune_qm9_gap.sh
diff --git a/finetune/run_finetune_qm9_h298.sh b/3dgrid_vqgan/finetune/run_finetune_qm9_h298.sh
similarity index 100%
rename from finetune/run_finetune_qm9_h298.sh
rename to 3dgrid_vqgan/finetune/run_finetune_qm9_h298.sh
diff --git a/finetune/run_finetune_qm9_homo.sh b/3dgrid_vqgan/finetune/run_finetune_qm9_homo.sh
similarity index 100%
rename from finetune/run_finetune_qm9_homo.sh
rename to 3dgrid_vqgan/finetune/run_finetune_qm9_homo.sh
diff --git a/finetune/run_finetune_qm9_lumo.sh b/3dgrid_vqgan/finetune/run_finetune_qm9_lumo.sh
similarity index 100%
rename from finetune/run_finetune_qm9_lumo.sh
rename to 3dgrid_vqgan/finetune/run_finetune_qm9_lumo.sh
diff --git a/finetune/run_finetune_qm9_mu.sh b/3dgrid_vqgan/finetune/run_finetune_qm9_mu.sh
similarity index 100%
rename from finetune/run_finetune_qm9_mu.sh
rename to 3dgrid_vqgan/finetune/run_finetune_qm9_mu.sh
diff --git a/finetune/run_finetune_qm9_r2.sh b/3dgrid_vqgan/finetune/run_finetune_qm9_r2.sh
similarity index 100%
rename from finetune/run_finetune_qm9_r2.sh
rename to 3dgrid_vqgan/finetune/run_finetune_qm9_r2.sh
diff --git a/finetune/run_finetune_qm9_u0.sh b/3dgrid_vqgan/finetune/run_finetune_qm9_u0.sh
similarity index 100%
rename from finetune/run_finetune_qm9_u0.sh
rename to 3dgrid_vqgan/finetune/run_finetune_qm9_u0.sh
diff --git a/finetune/run_finetune_qm9_u298.sh b/3dgrid_vqgan/finetune/run_finetune_qm9_u298.sh
similarity index 100%
rename from finetune/run_finetune_qm9_u298.sh
rename to 3dgrid_vqgan/finetune/run_finetune_qm9_u298.sh
diff --git a/finetune/run_finetune_qm9_zpve.sh b/3dgrid_vqgan/finetune/run_finetune_qm9_zpve.sh
similarity index 100%
rename from finetune/run_finetune_qm9_zpve.sh
rename to 3dgrid_vqgan/finetune/run_finetune_qm9_zpve.sh
diff --git a/finetune/trainers.py b/3dgrid_vqgan/finetune/trainers.py
similarity index 100%
rename from finetune/trainers.py
rename to 3dgrid_vqgan/finetune/trainers.py
diff --git a/finetune/utils.py b/3dgrid_vqgan/finetune/utils.py
similarity index 100%
rename from finetune/utils.py
rename to 3dgrid_vqgan/finetune/utils.py
diff --git a/images/3dgridvqgan_architecture.png b/3dgrid_vqgan/images/3dgridvqgan_architecture.png
similarity index 100%
rename from images/3dgridvqgan_architecture.png
rename to 3dgrid_vqgan/images/3dgridvqgan_architecture.png
diff --git a/inference/run_embeddings_eval_xgboost.sh b/3dgrid_vqgan/inference/run_embeddings_eval_xgboost.sh
similarity index 100%
rename from inference/run_embeddings_eval_xgboost.sh
rename to 3dgrid_vqgan/inference/run_embeddings_eval_xgboost.sh
diff --git a/inference/run_extract_embeddings.sh b/3dgrid_vqgan/inference/run_extract_embeddings.sh
similarity index 100%
rename from inference/run_extract_embeddings.sh
rename to 3dgrid_vqgan/inference/run_extract_embeddings.sh
diff --git a/inference/run_image_eval.sh b/3dgrid_vqgan/inference/run_image_eval.sh
similarity index 100%
rename from inference/run_image_eval.sh
rename to 3dgrid_vqgan/inference/run_image_eval.sh
diff --git a/inference/scripts/__init__.py b/3dgrid_vqgan/inference/scripts/__init__.py
similarity index 100%
rename from inference/scripts/__init__.py
rename to 3dgrid_vqgan/inference/scripts/__init__.py
diff --git a/inference/scripts/evaluate_embeddings_xgboost.py b/3dgrid_vqgan/inference/scripts/evaluate_embeddings_xgboost.py
similarity index 100%
rename from inference/scripts/evaluate_embeddings_xgboost.py
rename to 3dgrid_vqgan/inference/scripts/evaluate_embeddings_xgboost.py
diff --git a/inference/scripts/evaluate_image.py b/3dgrid_vqgan/inference/scripts/evaluate_image.py
similarity index 100%
rename from inference/scripts/evaluate_image.py
rename to 3dgrid_vqgan/inference/scripts/evaluate_image.py
diff --git a/inference/scripts/extract_embeddings.py b/3dgrid_vqgan/inference/scripts/extract_embeddings.py
similarity index 100%
rename from inference/scripts/extract_embeddings.py
rename to 3dgrid_vqgan/inference/scripts/extract_embeddings.py
diff --git a/inference/vq_gan_3d/__init__.py b/3dgrid_vqgan/inference/vq_gan_3d/__init__.py
similarity index 100%
rename from inference/vq_gan_3d/__init__.py
rename to 3dgrid_vqgan/inference/vq_gan_3d/__init__.py
diff --git a/inference/vq_gan_3d/dataset/__init__.py b/3dgrid_vqgan/inference/vq_gan_3d/dataset/__init__.py
similarity index 100%
rename from inference/vq_gan_3d/dataset/__init__.py
rename to 3dgrid_vqgan/inference/vq_gan_3d/dataset/__init__.py
diff --git a/inference/vq_gan_3d/dataset/default.py b/3dgrid_vqgan/inference/vq_gan_3d/dataset/default.py
similarity index 100%
rename from inference/vq_gan_3d/dataset/default.py
rename to 3dgrid_vqgan/inference/vq_gan_3d/dataset/default.py
diff --git a/inference/vq_gan_3d/metrics.py b/3dgrid_vqgan/inference/vq_gan_3d/metrics.py
similarity index 100%
rename from inference/vq_gan_3d/metrics.py
rename to 3dgrid_vqgan/inference/vq_gan_3d/metrics.py
diff --git a/inference/vq_gan_3d/model/__init__.py b/3dgrid_vqgan/inference/vq_gan_3d/model/__init__.py
similarity index 100%
rename from inference/vq_gan_3d/model/__init__.py
rename to 3dgrid_vqgan/inference/vq_gan_3d/model/__init__.py
diff --git a/inference/vq_gan_3d/model/cache/vgg.pth b/3dgrid_vqgan/inference/vq_gan_3d/model/cache/vgg.pth
similarity index 100%
rename from inference/vq_gan_3d/model/cache/vgg.pth
rename to 3dgrid_vqgan/inference/vq_gan_3d/model/cache/vgg.pth
diff --git a/inference/vq_gan_3d/model/codebook.py b/3dgrid_vqgan/inference/vq_gan_3d/model/codebook.py
similarity index 100%
rename from inference/vq_gan_3d/model/codebook.py
rename to 3dgrid_vqgan/inference/vq_gan_3d/model/codebook.py
diff --git a/inference/vq_gan_3d/model/lpips.py b/3dgrid_vqgan/inference/vq_gan_3d/model/lpips.py
similarity index 100%
rename from inference/vq_gan_3d/model/lpips.py
rename to 3dgrid_vqgan/inference/vq_gan_3d/model/lpips.py
diff --git a/inference/vq_gan_3d/model/vqgan.py b/3dgrid_vqgan/inference/vq_gan_3d/model/vqgan.py
similarity index 100%
rename from inference/vq_gan_3d/model/vqgan.py
rename to 3dgrid_vqgan/inference/vq_gan_3d/model/vqgan.py
diff --git a/inference/vq_gan_3d/utils.py b/3dgrid_vqgan/inference/vq_gan_3d/utils.py
similarity index 100%
rename from inference/vq_gan_3d/utils.py
rename to 3dgrid_vqgan/inference/vq_gan_3d/utils.py
diff --git a/training/config/base_cfg.yaml b/3dgrid_vqgan/training/config/base_cfg.yaml
similarity index 100%
rename from training/config/base_cfg.yaml
rename to 3dgrid_vqgan/training/config/base_cfg.yaml
diff --git a/training/config/dataset/default.yaml b/3dgrid_vqgan/training/config/dataset/default.yaml
similarity index 100%
rename from training/config/dataset/default.yaml
rename to 3dgrid_vqgan/training/config/dataset/default.yaml
diff --git a/training/config/model/vq_gan_3d.yaml b/3dgrid_vqgan/training/config/model/vq_gan_3d.yaml
similarity index 100%
rename from training/config/model/vq_gan_3d.yaml
rename to 3dgrid_vqgan/training/config/model/vq_gan_3d.yaml
diff --git a/training/dataset/__init__.py b/3dgrid_vqgan/training/dataset/__init__.py
similarity index 100%
rename from training/dataset/__init__.py
rename to 3dgrid_vqgan/training/dataset/__init__.py
diff --git a/training/dataset/default.py b/3dgrid_vqgan/training/dataset/default.py
similarity index 100%
rename from training/dataset/default.py
rename to 3dgrid_vqgan/training/dataset/default.py
diff --git a/training/run_mpi_training.sh b/3dgrid_vqgan/training/run_mpi_training.sh
similarity index 100%
rename from training/run_mpi_training.sh
rename to 3dgrid_vqgan/training/run_mpi_training.sh
diff --git a/training/train/callbacks.py b/3dgrid_vqgan/training/train/callbacks.py
similarity index 100%
rename from training/train/callbacks.py
rename to 3dgrid_vqgan/training/train/callbacks.py
diff --git a/training/train/get_dataset.py b/3dgrid_vqgan/training/train/get_dataset.py
similarity index 100%
rename from training/train/get_dataset.py
rename to 3dgrid_vqgan/training/train/get_dataset.py
diff --git a/training/train/train_vqgan_DDP.py b/3dgrid_vqgan/training/train/train_vqgan_DDP.py
similarity index 100%
rename from training/train/train_vqgan_DDP.py
rename to 3dgrid_vqgan/training/train/train_vqgan_DDP.py
diff --git a/training/train/trainer.py b/3dgrid_vqgan/training/train/trainer.py
similarity index 100%
rename from training/train/trainer.py
rename to 3dgrid_vqgan/training/train/trainer.py
diff --git a/training/vq_gan_3d/model/__init__.py b/3dgrid_vqgan/training/vq_gan_3d/model/__init__.py
similarity index 100%
rename from training/vq_gan_3d/model/__init__.py
rename to 3dgrid_vqgan/training/vq_gan_3d/model/__init__.py
diff --git a/training/vq_gan_3d/model/cache/vgg.pth b/3dgrid_vqgan/training/vq_gan_3d/model/cache/vgg.pth
similarity index 100%
rename from training/vq_gan_3d/model/cache/vgg.pth
rename to 3dgrid_vqgan/training/vq_gan_3d/model/cache/vgg.pth
diff --git a/training/vq_gan_3d/model/codebook.py b/3dgrid_vqgan/training/vq_gan_3d/model/codebook.py
similarity index 100%
rename from training/vq_gan_3d/model/codebook.py
rename to 3dgrid_vqgan/training/vq_gan_3d/model/codebook.py
diff --git a/training/vq_gan_3d/model/lpips.py b/3dgrid_vqgan/training/vq_gan_3d/model/lpips.py
similarity index 100%
rename from training/vq_gan_3d/model/lpips.py
rename to 3dgrid_vqgan/training/vq_gan_3d/model/lpips.py
diff --git a/training/vq_gan_3d/model/vqgan_DDP.py b/3dgrid_vqgan/training/vq_gan_3d/model/vqgan_DDP.py
similarity index 100%
rename from training/vq_gan_3d/model/vqgan_DDP.py
rename to 3dgrid_vqgan/training/vq_gan_3d/model/vqgan_DDP.py
diff --git a/training/vq_gan_3d/utils.py b/3dgrid_vqgan/training/vq_gan_3d/utils.py
similarity index 100%
rename from training/vq_gan_3d/utils.py
rename to 3dgrid_vqgan/training/vq_gan_3d/utils.py