I have trained CNNs using PyTorch, and I am trying to run a train loop on a simple CNN in Flux for binary (cat/dog) classification, but I cannot for the life of me figure out how to get the loss functions to work in Flux. I have looked around and copied what others did, but to no avail. It has to be something in the way I am using DataLoader, I just know it. But it could also be a collection of issues, so I am going to link the GitHub repo. The images are loading in properly, so I don’t think it’s that, but it could be how I am manipulating and treating them within the code.
Thank you for any help!