Skip to content

Rewrite training loop in PyTorch

Alexandru-Mihai GHERGHESCU requested to merge feature/pytorch into main

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.

Merge request reports