Update test_images.py

htyj123 · Nov 28, 2018 · 5fe132f · 5fe132f
1 parent ff4bb09
commit 5fe132f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test_images.py b/test_images.py
@@ -1,8 +1,8 @@
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
-from models.sa_gan_l2h_unet import InpaintRUNNet, InpaintSADirciminator
-from models.sa_gan import InpaintSANet
+#from models.sa_gan_l2h_unet import InpaintRUNNet, InpaintSADirciminator
+from models.sa_gan import InpaintSANet, InpaintSADirciminator
 from models.loss import SNDisLoss, SNGenLoss, ReconLoss, PerceptualLoss, StyleLoss
 from util.logger import TensorBoardLogger
 from util.config import Config