search

junyanz / VON

[NeurIPS 2018] Visual Object Networks: Image Generation with Disentangled 3D Representation.
http://von.csail.mit.edu
Other
530 stars 90 forks source link

Using CPU only #26

Closed hakanyi closed 4 years ago

hakanyi commented 4 years ago

I've been trying to run the pipeline (only testing, no training) using a CPU only. Was this repository intended to do so? In the readme it says that blender is optional and I interpreted it as blender being an alternative to the custom render module based on cuda. However, blender only seems to use whatever model.sample_2d() outputs and model.sample_2d() in turn is based on the custom render module. Thanks in advance for commenting on this!

junyanz commented 4 years ago

Not sure if it works for CPU, as the custom render module requires CUDA. The blender is NOT an alternative to the custom module. The blender is used for visualization, while the custom module is used for training.

ztzhang commented 4 years ago

I think the current version does not support CPU-only training/testing. The custom CUDA kernel is used during the generation process, therefore you have to use GPU for testing.

hakanyi commented 4 years ago

Thanks for clarifying!