NOTE: This is old Model class and is deprecated. It will no longer be maintained! Please use version > 0.5.1. Its much better and supports multi-gpu training too!

[jeremyhermann]

I have 0.7.2 installed via pip and I am getting the version error message. Is that expected?

$ pip list | grep tez
tez                        0.7.2

[abhishekkrthakur]

please see this example and convert your code to new Model class and get rid of the warning: https://github.com/abhishekkrthakur/tez/blob/main/examples/image/digit_recognizer.py :)