decodertest.py

def main():
    encoder = VGG19Encoder()
    decoder = Decoder()
    print(decoder)

    sample_input = torch.ones((1, 3, 256, 256))
    outputs = encoder(sample_input)
    output = decoder(outputs[-1])

    print(f"Input shape: {sample_input.shape}")
    print(f"Intermediate shapes: {[output.shape for output in outputs]}")
    print(f"Output shape: {output.shape}")