if anyone is curious, I figured out a solution in which I can simply define hyperparameters that I might not actually use depending on the value of num_layers. This works because of the limited size range, but curious to think about this more for higher dynamic ranges.