Function save_upsample

python/unet.py:90–94

(upsample, path)

    save_conv2d(downsample.op, path)

def save_upsample(upsample, path):
    pathlib.Path(path).mkdir(parents=True, exist_ok=True)

    # Save Conv2d instance
    save_conv2d(upsample.conv, os.path.join(path, 'conv')) 


def save_res_transformer_res(block, path):