search

justusschock / shapenet

PyTorch implementation of "Super-Realtime Facial Landmark Detection and Shape Fitting by Deep Regression of Shape Model Parameters" predicting facial landmarks with up to 400 FPS
https://shapenet.rtfd.io
GNU Affero General Public License v3.0
342 stars 59 forks source link

tf.train.Optimizer error #30

Closed MiniMinyi closed 4 years ago

MiniMinyi commented 4 years ago

When I have tensorflow v2.0 in my environment and import shapenet, the tf.train.Optimizer doesn't exist error occurred. Seems like it's a compatibility issue. I downgrade my tensorflow to version 1.14.0, I can still see this warning The name tf.train.Optimizer is deprecated. Please use tf.compat.v1.train.Optimizer instead.

justusschock commented 4 years ago

This is a PyTorch framework. You should not need Tensorflow at all