Cleanup docstrings in lightly/models/modules subpackage.

[guarin]

See #1216 for more information

[goelnaman66]

Hii, can you assign this to me?

[guarin]

Hi, yes of course :)

[Abhrajitdas02]

hii, can you ssign this issue to me?

[guarin]

Hi @Abhrajitdas02! Thanks for your interest in this issue. We only assign issues to one person at a time.

@goelnaman66 did you already have time to look into this?

[fadkeabhi]

Hey @guarin I would like to work on this issue as discussed here #1216. For some reason I did not receive the notification about it and I could not comment on it, can you assign it to me now. I have already started to work on it.

[ayush22iitbhu]

@guarin I see this issue has been open for more than a week. I have already started working on this. Also, I have already completed the work you assigned to me on documentation of lightly/models/utils.py and lightly/loss subpackage. Please assign this issue to me. I'll complete this soon as well.

[guarin]

Hi @fadkeabhi and @ayush22iitbhu could you post which files you already updated? Then we can avoid that you work both on the same thing.

[ayush22iitbhu]

@guarin I have already updated center.py, heads.py, heads_timm.py and currently working on ijepa.py. But you can assign more files to me.

[guarin]

Could you already make a PR for these files? Then we can merge and avoid conflicts later on. The fewer files a PR touches the easier it is to merge for us :)

[ayush22iitbhu]

@guarin review PR