Zasder3 / train-CLIP

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

Update README.md #16

Closed talpay closed 3 years ago

talpay commented 3 years ago

Fixed example (there is no batchsize argument)

Zasder3 commented 3 years ago

That flag originates from Lightning's Trainer. I grab it's argparse arguments during initialization.