First, set self.scale=2.
Second, make adjustment to forward(self, x).
Only one tf.depth_to_space() is required for 2x.
convmerge1=Conv2D(12, 3, strides=ds, padding='same', activation=activate, kernel_initializer=ki, name='convmerge1') merge=tf.concat(inp0,axis=-1) merge=convmerge1(merge) out=tf.depth_to_space(merge,2)
First, set self.scale=2. Second, make adjustment to forward(self, x). Only one tf.depth_to_space() is required for 2x.
convmerge1=Conv2D(12, 3, strides=ds, padding='same', activation=activate, kernel_initializer=ki, name='convmerge1') merge=tf.concat(inp0,axis=-1) merge=convmerge1(merge) out=tf.depth_to_space(merge,2)