fixed typo in models

amanchadha · Dec 3, 2019 · c11c32d · c11c32d
1 parent d753b40
commit c11c32d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/FRVSRGAN_Models.py b/FRVSRGAN_Models.py
@@ -137,7 +137,7 @@ def __init__(self, batch_size, lr_height, lr_width):
         self.height = lr_height
         self.batch_size = batch_size
         self.fnet = FNet()
-        self.todepth = SpaceToDepth(FRVSRGAN.SRFactor)
+        self.todepth = SpaceToDepth(FRVSR.SRFactor)
         self.srnet = SRNet(FRVSR.SRFactor * FRVSR.SRFactor * 3 + 3)  # 3 is channel number
 
     # make sure to call this before every batch train.