search

Zasder3 / train-CLIP

A PyTorch Lightning solution to training OpenAI's CLIP from scratch.
MIT License
653 stars 78 forks source link

About Gradient Checkpointing #2

Open ChawDoe opened 3 years ago

ChawDoe commented 3 years ago

I wrote the codes about gradient checkpointing. Can I contribute some codes to this project?

Zasder3 commented 3 years ago

That would be amazing! Could send a pull request with your implementation?

ChawDoe commented 3 years ago

Okay, I will send it as soon as possible.