Skip to main content
NVIDIA Deep Learning Examples
Wide & Deep for TensorFlow2
Resource
NVIDIA Deep Learning Examples
Wide & Deep for TensorFlow2

Wide & Deep Recommender model.

  • Changelog

    June 2022

    • Added support for BYO Dataset

    May 2022

    • Added multi-hot categorical features
    • Added triton inference
    • Updated model architecture figure
    • Updated NVTabular to v0.7.1
    • Updated readme numbers

    November 2021

    • Refresh release with performance optimizations
    • Updated NVTabular to v0.6.1
    • Replaced native TF dataloader with NVTabular counterpart
    • Removed spark CPU preprocessing
    • Updated readme numbers
    • Changed V100 cards from 16GB to 32GB

    February 2021

    • Initial release

    Known issues

    • In this model the TF32 precision can in some cases be as fast as the FP16 precision on Ampere GPUs. This is because TF32 also uses Tensor Cores and doesn't need any additional logic such as maintaining FP32 master weights and casts. However, please note that W&D is, by modern recommender standards, a very small model. Larger models should still have significant benefits of using FP16 math.

    Modal Content

    NVIDIA uses cookies to improve your experience on our web site. We and our third-party partners also use cookies and other tools to collect and record information you provide as well as information about your interactions with our websites for performance improvement, analytics, and to assist in marketing efforts. By clicking "Accept All", you consent to our use of cookies and other tools as described in our Cookie Policy. You can manage your cookie settings by clicking on "Manage Settings." By continuing to use this site or by clicking one of the buttons below, you agree to our Terms of Service (which contains important waivers). Please see our Privacy Policy for more information on our privacy practices.