microsoft / SpareNet

Style-based Point Generator with Adversarial Rendering for Point Cloud Completion (CVPR 2021)

Home Page:https://arxiv.org/abs/2103.02535

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Batch Time is very slow (around 4 sec) with 4 GPUs (1080ti)

coreqode opened this issue · comments

image

image

BatchTime is very slow. I am using batch size of 12 as of now.

We use 4 Tesla V100 GPUs to train our model. With the batch size 32, it takes an average of 1.2 seconds to finish one batch. During the training, we need to calculate EMD loss and also render depth map, so the training process can be time-consuming.