Out of memory during Inference

simpleig / Geo-PIFu

This repository is the official implementation of Geo-PIFu: Geometry and Pixel Aligned Implicit Functions for Single-view Human Reconstruction.

112 stars 17 forks source link

Out of memory during Inference #25

Open SnehithR opened 3 years ago

SnehithR commented 3 years ago

I'm using a GTX 1080 Ti (11GB Memory) for the testing of the given model with the pre-trained checkpoints. When I run the test_shape_iccv it displays CUDA out of memory. Does the model require more than 11 GB of memory for infernece?

Thanks in Advance.

direcf commented 3 years ago

I am on the same page