NVIDIA-Merlin / models

Merlin Models is a collection of deep learning recommender system model reference implementations

Apache License 2.0

251 stars 50 forks source link

Open bschifferer opened 1 year ago

bschifferer commented 1 year ago

🚀 Feature request

rnyak commented 1 year ago

@bschifferer did you get a chance to test these two lines and see some speed up?

policy = mixed_precision.Policy('mixed_float16')
mixed_precision.set_global_policy(policy)