From 2636d3b6208417dd275315e58ee8037440648616 Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 6 Jul 2020 09:23:53 -0600 Subject: [PATCH] Fix assertion error --- codes/models/archs/SRG1_arch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/archs/SRG1_arch.py b/codes/models/archs/SRG1_arch.py index 826ccd58..c1af2836 100644 --- a/codes/models/archs/SRG1_arch.py +++ b/codes/models/archs/SRG1_arch.py @@ -66,7 +66,7 @@ class SwitchComputer(nn.Module): tc = transform_count if self.enable_negative_transforms: tc = transform_count * 2 - assert filters > transform_count * 2 + assert filters > tc self.final_switch_conv = nn.Conv2d(filters, tc, 1, 1, 0) self.transforms = nn.ModuleList([transform_block() for _ in range(transform_count)])