pediamedai / vitasd Goto Github PK

I tried several times to run the complete code as per your instructions, but unable to run it. Some functions are missing. Can you provide me with the complete code?
It would be very helpful.

In 6th line of train_affectnet.py,
from models import ViTASD, ResNetASD
where ResNetASD is missing.

Looks like a bad version was uploaded

File "/content/ViTASD/train_affectnet.py", line 72, in init
self.model: torch.nn.Module = ResNetASD(
NameError: name 'ResNetASD' is not defined

When I run the code I get the following error, how do I fix it?

In 7th line of train.py, from lib.pos_embed import interpolate_pos_embed, where lib.pos_embed is missing.

I have successfully run your code. But the accuracy is very different from your paper.
What should I use as a pre-trained model path in pretrain_path: str = "" in train_affectnet.py line no: 40 ?

error: Validation failed: No action for key "trainer.weights_save_path" to check its value.

Hi, I hope you are doing well. I wanted to first congratulate you on this well maintained repository. I had a little question regarding pretraining the ViT on affectnet dataset. Does this repository only contains the code for pretraining the ViT on Affectnet in supervised way?