Sure. Let’s first see if we can scale down the problem.
Would it be sufficient, if we just use BottleNeck modules or du you need the flexibility of BasicBlock/BottleNeck?
Also, do you need the different ResNet architectures, i.e. resnet18, resnet34, ... or can we focus on just one implementation?