Merge pull request #347 from alan-turing-institute/vit-mae-pretrain #857

Job	Run time
Test on windows-latest with Python 3.12	5m 2s
Test on macos-latest with Python 3.11	7m 5s
Test on ubuntu-latest with Python 3.12	2m 37s
Test on windows-latest with Python 3.11	4m 4s
Test on ubuntu-latest with Python 3.11	2m 26s
Test on macos-latest with Python 3.12	5m 22s
	26m 36s

Provide feedback