How much memory is needed to run this code?

DSaurus / Tensor4D

This is the official implementation of Tensor4D: Efficient Neural 4D Decomposition for High-fidelity Dynamic Reconstruction and Rendering.

MIT License

191 stars 4 forks source link

Open RuoyiDu opened 1 year ago

RuoyiDu commented 1 year ago

Excellent work! How much memory is needed to run this code (train and inference)?

DSaurus commented 1 year ago

Hi @RuoyiDu,

It requires ~15GB of memory in most cases. You can also decrease batch_size or use fp16 mode to further reduce memory usage.