Vision-Transformer-ViT

Vision-Transformer-ViT copied to clipboard

`total_steps` in train.py

1

There seems to be a problem in train.py. ``` total_steps = (len(trainloader) // args.batch_size + 1) * args.epoches ``` len(train_loader) is already divided by batch_size. Change it to `total_steps =...

JJJYmmm