tensorflow / compression

Data compression in TensorFlow
Apache License 2.0
860 stars 251 forks source link

Update gdn.py to support mixed precision training #101

Closed MahmoudAshraf97 closed 3 years ago

MahmoudAshraf97 commented 3 years ago

an error rises when using mixed precision that the convert_to_tensor calls are receiving tensor inputs, tf.cast solved the problem