tensorflow.contrib error in train.py

mks0601 / TF-SimpleHumanPose

TensorFlow implementation of "Simple Baselines for Human Pose Estimation and Tracking", ECCV 2018

336 stars 90 forks source link

Closed Dipankar1997161 closed 2 years ago

Dipankar1997161 commented 2 years ago

I am using TensorFlow 2.8 and python 3.8.10 in conda env. As I was running the train.py, I came across this error.

I guess, in TensorFlow 2.8, contrib is not there.

So what can be the possible solution here?

Should I downgrade my TensorFlow and python? If yes then which versions??

Dipankar1997161 commented 2 years ago

I was able to downgrade it to tensorflow-gpu= 1.15 python = 3.6.13, Cudatoolkit=10.0 Cudnn=7.2.1

My training has started!! Hope I get some results Thank you