It seems you are using a higher-level API, which provides a fit and predict function, and which might use automatic mixed precision internally.
Which library are you using and how did you specify to use amp?