luni/cvpr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update pyramid

Browse Source
This commit is contained in:
luni
2022-10-16 14:57:36 +00:00
parent a6834caf80
commit 07c9fb57df
3 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/pyramid_level.py
+2 -2
View File
@@ -9,7 +9,7 @@ import scipy as sp
from custom_utils.custom_utils import load_model
from models.base import AEEncoder, VariationalEncoder, Decoder
from models.upscaler import UpscalerBasic, UpscalerBasicNew
from models.upscaler import UpscalerBasic, UpscalerBasicConv, UpscalerBasicNew
class PyramidLevel(tf.keras.Model):
@@ -24,7 +24,7 @@ class PyramidLevel(tf.keras.Model):
self.upscale_enhancer = None
if config['upscale_enhancer'] != '':
self.upscale_enhancer = UpscalerBasicNew(config['resolution'])
self.upscale_enhancer = UpscalerBasicConv(config['resolution'])
load_model(self.upscale_enhancer, config['upscale_enhancer'])
self.upscale_enhancer.trainable = not config['upscale_enhancer_freeze']