Function save_encoder

python/autoencoder.py:72–83

(encoder, path)

        save_padded_conv2d(encoder_block['downsample']['conv'], pathlib.Path(path, 'downsampler'))

def save_encoder(encoder, path):
    pathlib.Path(path).mkdir(parents=True, exist_ok=True)
    
    save_conv2d(encoder.conv_in, pathlib.Path(path, 'conv_in'))
    save_mid(encoder.mid, pathlib.Path(path, 'mid'))

    for i, block in enumerate(encoder.down):
        save_encoder_block(block, pathlib.Path(path, f'blocks/{i}'))

    save_scalar(len(encoder.down), "n_block", path)
    save_group_norm(encoder.norm_out, pathlib.Path(path, 'norm_out'))
    save_conv2d(encoder.conv_out, pathlib.Path(path, 'conv_out'))
    

def save_autoencoder(autoencoder, path):