EfficientPose icon indicating copy to clipboard operation
EfficientPose copied to clipboard

Published 20 hours ago • verified publisher icon ybkscht

Tensorflow2 support

Open mrunalsompura1995 opened this issue 2 years ago • 2 comments

Hello Everyone.

I am trying to run EfficientPose on RTX A4000 GPU, it works fine with GTX series but with higher CUDA versions, I guess tensorflow 1.15 is not sufficient. It needs to be run with tensorflow2 to get advantage of computation power.

If there is support for tensorflow2 or any pytorch implementation of EfficientPose it would be great help.

Thanks in advance.

mrunalsompura1995 avatar Aug 15 '22 12:08 mrunalsompura1995

I had the same problem with RTX 3090. I modified it to version2, but i think there's a problem with training.

jusepv avatar Nov 23 '22 01:11 jusepv

Hi @jusepv

Have you been able to get it working with Tensorflow 2? Specifically, the inference, as the training is not an issue.

Kaladin-Syl-WR avatar Jun 08 '23 11:06 Kaladin-Syl-WR