mbanani / lgssl Goto Github PK

Hi,
Firstly, congratulations on the good work, and making your project publicly available.

When setting up the project with the dependencies in the requirements.txt file, some modules do not have their versions mentioned, and the code does not seem to be forward compatible (with the latest versions). For instance pytorch-lightning, etc.

Would be great if you could also update that, since it took a lot of effort to finally get the current version up and running.

Many Thanks :)

P.S.: Maybe even a pip-freeze of your conda / virtualenv would be really helpful :D

Hello, I found that the get_dataset and subsample_dataset functions are missing in the sample_language_nn.py file when I executed your code. Could you please provide them?

Thanks for sharing the code!

Is there a way to load the model and perform inference inside the code like

model = load_lgssl()

data = my_data[0] # single example
results = model(data)