Upgrading tf.contrib.slim to tf 2.0

yzcjtr / GeoNet

Code for GeoNet: Unsupervised Learning of Dense Depth, Optical Flow and Camera Pose (CVPR 2018)

MIT License

723 stars 181 forks source link

Upgrading tf.contrib.slim to tf 2.0 #65

Closed essamgoda closed 4 years ago

essamgoda commented 4 years ago

I have a problem with my the code which uses tf.contrib.slim functionalities and it does not work with TensorFlow to 2.0. How can I upgrade?

yzcjtr commented 4 years ago

Sorry I haven't used tf for quite a long time. I remember most of the slim ops are related with common modules to build the network architecture (such as conv, avg_pool). Their counterparts should be easy to find in tf2.0, though some detailed discrepancy might exist in the implementation.

yzcjtr commented 4 years ago

Also, I find sonnet (https://github.com/deepmind/sonnet) is a good replacement. Hope it helps.