From 69da86bd56fe09860b0a2c0968e1ce75845ebba2 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Wed, 25 Dec 2019 15:34:00 -0400 Subject: [PATCH] next --- pix2pixHD_attack/test.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pix2pixHD_attack/test.py b/pix2pixHD_attack/test.py index d9f208c..94c8419 100755 --- a/pix2pixHD_attack/test.py +++ b/pix2pixHD_attack/test.py @@ -8,6 +8,7 @@ import util.util as util from util.visualizer import Visualizer from util import html import torch +import torch.nn.functional as F opt = TestOptions().parse(save=False) opt.nThreads = 1 # test code only supports nThreads = 1