3D Sparse Voxel Autoencoder
Inspired by the machine learning research I did, I created a 3D autoencoder model using the spconv framework with PyTorch. The key innovation is training a sparse model, because most of a voxelized model is empty. This massively decreases VRAM usage and increases training speed.
Figure 1: Model Architecture.