aljaz/InvSR
1
0
Fork 0
mirror of https://github.com/aljazceru/InvSR.git synced 2025-12-17 06:14:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

add chopping bs in argparse

Browse Source
This commit is contained in:
zsyOAOA
2024-12-13 15:20:00 +08:00
parent f980a9df14
commit f138fdff1b
3 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
sampler_invsr.py
View File

@@ -173,7 +173,7 @@ class InvSamplerSR(BaseSampler):
stride= int(idle_pch_size * 0.50),
sf=self.configs.basesr.sf,
weight_type=self.configs.basesr.chopping.weight_type,
extra_bs=1 if self.configs.bs > 1 else self.configs.basesr.chopping.extra_bs,
extra_bs=self.configs.basesr.chopping.extra_bs,
)
for im_lq_pch, index_infos in im_spliter:
target_size = (