Step4： No module named fairscale.nn

[witty1011]

Hi, author. Thanks for ur great job. I tried to test my dataset on ur modle but that error 'No module named fairscale.nn' happened when I forward to step 4. Tried to reconda and reinstall fairscale but all failed. Did same error happen when u run?

[lucazanella]

Hi, thanks for your interest in our work!
I think I experienced the same issue before. Could you try activating the virtual environment and then manually installing fairscale with the command:
pip install fairscale
Let me know if this helps!