4a5eef64d2a0ab5e1737ffbb345d9e72d2f1adc9,test.py,,,#,48

Before Change

                style = coral(style, content)
            style = style.cuda()
            content = content.cuda()
            content = Variable(content.unsqueeze(0), volatile=True)
            style = Variable(style.unsqueeze(0), volatile=True)

            output = style_transfer(vgg, decoder, content, style,

After Change

            style = style_tf(Image.open(style_path))
            if args.preserve_color:
                style = coral(style, content)
            style = style.to(device).unsqueeze(0)
            content = content.to(device).unsqueeze(0)
            with torch.no_grad():
                output = style_transfer(vgg, decoder, content, style,

Instances