search

uw-mad-dash / bagpipe

Code for reproducing results for SOSP paper Bagpipe
MIT License
8 stars 2 forks source link

Unable to get torchrec_baseline #2

Open sherlockkenan opened 2 months ago

sherlockkenan commented 2 months ago

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 commented 2 months ago

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 commented 2 months ago

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 ?