Function save_autoencoder

python/autoencoder.py:86–92

(autoencoder, path)

    

def save_autoencoder(autoencoder, path):
    pathlib.Path(path).mkdir(parents=True, exist_ok=True)

    save_encoder(autoencoder.encoder, pathlib.Path(path, 'encoder'))
    save_decoder(autoencoder.decoder, pathlib.Path(path, 'decoder'))
    save_conv2d(autoencoder.quant_conv, pathlib.Path(path, 'quant_conv'))
    save_conv2d(autoencoder.post_quant_conv, pathlib.Path(path, 'post_quant_conv'))