Leveling up Training: NVTabular and PyTorch Lightning

Web: https://towardsdatascience.com/leveling-up-training-nvtabular-and-pytorch-lightning-57faf6242c11?source=rss----7f60cf5620c9---4

Training a wide and deep recommender model on MovieLens 25M

NVTabular is a feature engineering framework designed to work with NVIDIA Merlin. It can process large datasets typical in production recommender setups. I tried to work with NVIDIA Merlin on free instances, but the recommended approach seems to be the only way forward. But I still wanted to use NVTabular since the value of using the GPU for data engineering and data loading is very attractive. In this post, …

deep learning machine learning nvidia pytorch pytorch lightning recommendation-system training