Skip to content

Rewrite training loop in PyTorch

Alexandru-Mihai GHERGHESCUrequested 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