Rewrite training loop in PyTorch

Rewrite the current training loop in pytorch. This avoids relying too much on frameworks such as fastai or hugging face and gives us a chance to learn what the ... is happening.

requested review from @vlad_andrei.badoiu1

approved this merge request

merged

mentioned in commit 17c2eda3