ViT-pytorch

ViT-pytorch copied to clipboard

Could you please provide me a test codes? !@jeonsworld @jeonsworld

longmalongma

Could you please also provide a test.py file

2

thanks a million.

xiaohong1

how to deal with video，which has 5min length

1

dotsonliu

Not able to load ViT-H_14

1

I was testing using the provided visualize_attention_map.ipynb The ViT-B_16-224 loads fine but when I downloaded and was loading ViT-H_14, it gave me the following error: ``` RuntimeError Traceback (most recent...

abhijay9

Here in the if condition [code](https://github.com/jeonsworld/ViT-pytorch/blob/main/train.py#L206), is check for (step+1) == len(trainset) missing? Reference --> [Reference code](https://kozodoi.me/python/deep%20learning/pytorch/tutorial/2021/02/19/gradient-accumulation.html)

animesh-007

Hi, thanks for releasing this code. I have tried to run the CIFAR-10 (as well as CIFAR-100) example, but in both cases the validation (and training) loss do not decrease,...

josedolz

make a quick test file, might be of some use?

zanehartley

fix typo in comment

topsy404

Maybe we should decrease the batch size by the value of `--train_batch_size` if We do not have enough memory?

topsy404

Why the model gives the same logits for both the classes?

2

Hi, I am using ViT-H_14 pre-trained to perform binary classification of biomedical images. The dataset I have available is very small: I use about 300 images to perform fine tuning...

evapachetti