Challenges in Multi-Task Learning

I am working on a very challenging histopathology project for cancer detection. I am trying to train a model that not only predicts the cancer, but also ten associated mutations with it. Currently, I am facing a batch error.

I really don’t know much of what I am doing, and I am throwing together a patchwork that will throw many more errors. I don’t know much about the core of how PyTorch works. I really hope someone can see this topic and help me out…

Here is my project file. If you can provide any assistance, they will be greatly appreciated. Here is the link to my notebook: