Giters

jeonsworld / ViT-pytorch

Pytorch reimplementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Missing check of when (step+1) == len(trainset) in gradient_accumulation

animesh-007 opened this issue · comments

commented

Here in the if condition code, is check for (step+1) == len(trainset) missing?

Reference --> Reference code