github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-08 05:51:40 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

SAEHD: fix hd arhis

Browse source
This commit is contained in:
Colombo 2020-02-25 21:58:05 +04:00
parent b3b97b9802
commit a1cc297cef
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
models/Model_SAEHD/Model.py
View file

@ -283,6 +283,9 @@ class SAEHDModel(ModelBase):
weights = self.upscale0.get_weights() + self.upscale1.get_weights() + self.upscale2.get_weights() \
+ self.res0.get_weights() + self.res1.get_weights() + self.res2.get_weights() + self.out_conv.get_weights()
if self.is_hd:
weights += self.res3.get_weights()
if include_mask:
weights += self.upscalem0.get_weights() + self.upscalem1.get_weights() + self.upscalem2.get_weights() \
+ self.out_convm.get_weights()