search

jeonsworld / ViT-pytorch

Pytorch reimplementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale)
MIT License
1.95k stars 374 forks source link

How to set the number of epoch? #40

Closed xiaoweidao closed 3 years ago

xiaoweidao commented 3 years ago

How to set the number of epoch?