torchrec
torchrec copied to clipboard
pytorch
Pytorch domain library for recommendation systems
Summary: * add forward overide. * adapt length tensor shape for different ops. * insert per run context. Differential Revision: D39710186
Inside the train_pipeline.py `progress` call ``` File ""/opt/ee/python/3.8/lib/python3.8/site-packages/torchrec/distributed/train_pipeline.py"", line 460 in _connect ``` I will sometimes get the following error ``` "textPayload": "*** Error in `/opt/ee/python/3.8/bin/python': free(): invalid next size...
Hello, I am wondering how to migrate my model from _tensorflow_recommenders_ to pytorch _torchRec_, but I couldn't find the equivalent of the Retrival tasks and bruteForce predictions in the documentation....
Summary: Added INT8 support to torchrec comm library based on the changes in D39473386 Differential Revision: D39473544
Summary: # Problem Identified for the CMF model on TorchRec, its feature processors were not checkpointed. # Solution Add a unit test to guard its behavior Reviewed By: ge0405 Differential...
…ning set dataloading
Summary: Since applying an overlapped optimizer to all EBCs and ECs in a model is a common use case, creating a helper function `apply_overlapped_optimizer_to_module` doing just that for convenience. Differential...
Summary: ATT Differential Revision: D39581556
Summary: No need to expose create_context() to users. Differential Revision: D39319513
