I’m using the Finetuning Torchvision Models as a template, I hope this google colab notebook can be of help
https://colab.research.google.com/drive/175uIbG6u6DNdMmbbmfI8KW6R-HBm7hGf 6
The folder structure is
plane
–train
----plane1
----…
----plane70
–val
----plane1
----…
----plane70