Check out this video where we cover setting up the Burn training loop and dataloader for our custom machine learning model!