I see, thanks.
Related q, although maybe this deserves a new thread…what is this about?
if groups != 1 or base_width != 64:
  raise ValueError('BasicBlock only supports groups=1 and base_width=64')

Can I not adjust the width of the resnet? Is there a non-basic block I should be using?
I don’t care about width, per-say: I really just want a single parameter that I can adjust which will change the “scale” of the model, making it larger or smaller. What’s the best way to do this?