def save_output(gen, dest): out = gen.get_outputs()[0] io.imsave(dest, postprocess_img(out.asnumpy()[0]))