423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-02 03:42:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

install trainer inside pytorch3d

Browse Source 
Summary: One way to tidy the installation so we don't install files in site-packages/projects. Fixes https://github.com/facebookresearch/pytorch3d/issues/1279

Reviewed By: shapovalov, davnov134

Differential Revision: D38426772

fbshipit-source-id: ac1a54fbf230adb53904701e1f38bf9567f647ce
This commit is contained in:
Jeremy Reizenstein 2022-08-05 02:49:42 -07:00 committed by Facebook GitHub Bot
parent a0f786f4cf
commit 9d888f1332
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
setup.py
View File

@ -128,7 +128,7 @@ if os.getenv("PYTORCH3D_NO_NINJA", "0") == "1":
else: else:
BuildExtension = torch.utils.cpp_extension.BuildExtension BuildExtension = torch.utils.cpp_extension.BuildExtension
trainer = "projects.implicitron_trainer" trainer = "pytorch3d.implicitron_trainer"
setup( setup(
name="pytorch3d", name="pytorch3d",
@ -138,8 +138,10 @@ setup(
description="PyTorch3D is FAIR's library of reusable components " description="PyTorch3D is FAIR's library of reusable components "
"for deep Learning with 3D data.", "for deep Learning with 3D data.",
packages=find_packages( packages=find_packages(
exclude=("configs", "tests", "tests.*", "docs.*", "projects.nerf.*") exclude=("configs", "tests", "tests.*", "docs.*", "projects.*")
), )
+ [trainer],
package_dir={trainer: "projects/implicitron_trainer"},
install_requires=["fvcore", "iopath"], install_requires=["fvcore", "iopath"],
extras_require={ extras_require={
"all": ["matplotlib", "tqdm>4.29.0", "imageio", "ipywidgets"], "all": ["matplotlib", "tqdm>4.29.0", "imageio", "ipywidgets"],