A issue in python demo.py

DenisTome / Lifting-from-the-Deep-release

Implementation of "Lifting from the Deep: Convolutional 3D Pose Estimation from a Single Image"

https://denistome.github.io/papers/lifting-from-the-deep

GNU General Public License v3.0

450 stars 132 forks source link

A issue in python demo.py #27

Closed Huang9495 closed 6 years ago

Huang9495 commented 6 years ago

I operate the code python demo.py while a issue is occuring. And the issue as fellows:

My environment: python 2.7; GPU:1060 &1080ti; ubuntu16.04

Can you help me?

screenshot from 2018-09-22 08-13-13

DenisTome commented 6 years ago

Your specs are ok. They should not be the problem 1080ti can definitely handle our model. To further test this, you could reduce the batch size. To do that, edit the file packages/lifting/utils/config.py setting BATCH_SIZE = 1.

Huang9495 commented 6 years ago

@DenisTome OK, I try to do it. Thanks you