How to do installations correctly on Google Colab?

[HripsimeS]

@AlphaPav Hello. I wanted to see if I can train and test my dataset with your sparenet model on Google Colab, but unfortunately something went wrong and I get errors. You can find below the steps I did.

!git clone https://github.com/microsoft/SpareNet.git
%cd /content/SpareNet
!pip install -r requirements.txt
!sh setup_env.sh

The installations went fine without errors, but I get No module named 'emd' error while trying
from utils.misc import set_logger

I was assuming probably the setup_env.sh installation I did wrong. Can you please advice me how I can properly do all installations to be able to use train and test scrips on Google Colab. Thanks a lot in advance!!!