Unable to get torchrec_baseline

[sherlockkenan]

Hi,

I recently read your paper and found the idea very interesting!

I would like to compare it with TorchRec. However, I couldn't find the file ec2_launcher_six_models_torchrec_baseline.py mentioned in the README. Could you please provide the torchrec_baseline file?

Thank you!

[iidsample]

Thanks for your interest in our work.

I believe here is the file you might be looking for - https://github.com/uw-mad-dash/bagpipe/blob/main/ec2_launcher_six_models_baseline.py

[sherlockkenan]

Thank you for your response !

However, in the provided scripts, I only found the commands and instructions for running baseline models with Bagpipe. I did not find any information regarding TorchRec, such as how to run it or any model implementations. Is there anything I missed ?