distributed-embeddings
distributed-embeddings copied to clipboard
NVIDIA-Merlin
distributed-embeddings is a library for building large embedding based models in Tensorflow 2.
Synthetic Model Single GPU Example always gets OOM, even I use a A100 machine and set batch_size=1 python main.py --model small --optimizer sgd --batch_size 1 ``` root@2437d34894a8:/workspaces/distributed-embeddings/examples/benchmarks/synthetic_models# python main.py --model...
Current serialize/deserialize code in init is not neccessay
Currently if more input is supplied, only leading ones will be used and rest will be silently ignored. Need better check and message
All Merlin projects are switching from semantic versioning, like `0.4.0`, to calendar versioning, like `23.01`. The change is planned for January 2023. Add a banner to all doc pages to...
Metadata
Owner
Metadata
distributed-embeddings is a library for building large embedding based models in Tensorflow 2.