Hi Adam
There’s a fair bit of code here. Are you experiencing any errors or anything you weren’t expecting?
From a quick look through, it looks as though you have a single model (Net) that contains two separate ResNet models. So it sounds as though you may want to pass two datasets into the same model using one dataloader? If so, following something like this will be helpful.
Let me know how you get on or if I’ve misunderstood.