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

option to avoid accelerate

Browse Source 
Summary: For debugging, introduce PYTORCH3D_NO_ACCELERATE env var.

Reviewed By: shapovalov

Differential Revision: D37885393

fbshipit-source-id: de080080c0aa4b6d874028937083a0113bb97c23
This commit is contained in:
Jeremy Reizenstein
2022-07-17 13:15:59 -07:00
committed by Facebook GitHub Bot
parent 0f966217e5
commit 9b2e570536
2 changed files with 41 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
projects/implicitron_trainer/visualize_reconstruction.py
View File

@@ -344,7 +344,7 @@ def export_scenes(
os.environ["CUDA_VISIBLE_DEVICES"] = str(config.gpu_idx)
# Load the previously trained model
model, _, _ = init_model(config, force_load=True, load_model_only=True)
model, _, _ = init_model(cfg=config, force_load=True, load_model_only=True)
model.cuda()
model.eval()