search

faustomorales / vit-keras

Keras implementation of ViT (Vision Transformer)
Apache License 2.0
336 stars 80 forks source link

update mixed precision compatibility #12

Closed Shiro-LK closed 3 years ago

Shiro-LK commented 3 years ago

Hi,

This pull request is to add the compatibility with mixed precision. We need to cast in some call function few variables in order to make the mixed precision possible.

faustomorales commented 3 years ago

Thanks very much! This looks great.