enable using tf weight #3

Closed taiwen97 closed 3 years ago