How can I make conditional model architectures?
I’m writing a GAN and currently I have two classes defined as:
class Generator(nn.Module):
    ...
class Discriminator(nn.Module):
    ...

but I want to have multiple architectures dependent on the size of my input eg:
if len(input) == 16:
    # Use 16x16 Discriminator and Generator

if len(input) == 64:
    # Use 64x64 Discriminator and Generator

...

Is there a best-practice way to do this? I could define classes for each of my inputs but that seems a bit inefficient.