callummcdougall / ARENA_2.0

Resources for skilling up in AI alignment research engineering. Covers basics of deep learning, mechanistic interpretability, and RL.

190 stars 78 forks source link

execise.ipynbs need Lightning in colab #20

Closed matthiasdellago closed 1 year ago

matthiasdellago commented 1 year ago

In exercises 1.1 and 1.2 (and probably more) running in colab results in : `ModuleNotFoundError: No module named 'pytorch_lightning'

This can be fixed by including:

if IN_COLAB:

Install packages

%pip install pytorch-lightning

I could't find the exercise.ipynb files in the repo, if you can point me towards them I'll also gladly submit a PR.

P.S.: Thanks for providing this awesome content! :)

callummcdougall commented 1 year ago

No problem, glad you're enjoying it!

If this is referring to the first 2 exercise sets in the transformers material, I think they should both be fixed now? There shouldn't be any notebooks in the repo (I develop the exercises in notebooks but these are ignored in the "gitignore" file; they're just for development rather than for use). Let me know if there are still any issues though!

matthiasdellago commented 1 year ago

Ah, I should have specified: I'm talking about this notebook linked from this streamlit page . I think the other notebooks have the same problem.

callummcdougall commented 1 year ago

Ah, I see the issue - I'd removed the lightning install from the colabs, but hadn't quite removed it from all of the files in the repo. Have done so now, everything seems to be working. Thanks!