Pytorch код обучения

Взять любую из XoFTR или LoFTR, убрать lightning и запустить обучение. По итогу нужен код обучения без lightning на чистом pytorch.