Function save_mid

python/autoencoder.py:27–32

(mid, path)

    save_conv2d(attn_block.proj_out, pathlib.Path(path, 'proj_out'))

def save_mid(mid, path):
    pathlib.Path(path).mkdir(parents=True, exist_ok=True)

    save_resnet_block(mid.block_1, pathlib.Path(path, 'block_1'))
    save_attn_block(mid.attn_1, pathlib.Path(path, 'attn'))
    save_resnet_block(mid.block_2, pathlib.Path(path, 'block_2'))

def save_decoder_block(decoder_block, path):
    pathlib.Path(path).mkdir(parents=True, exist_ok=True)