Usage of training

[RegniesAnthony]

Hi,

I try to use deep learning with ViT for my project to localise nest in forest drone images. So I need to train my own model with my images, but I can't find out how to make it work.

my data :

when I try to use simply distributed_train.sh. I got this error in train.py :
UnboundLocalError : local variable 'input' referenced before assignment

And I can't understand why.

Thank you in advance.