Update src/streamlit_app.py
Browse files- src/streamlit_app.py +1 -1
src/streamlit_app.py
CHANGED
@@ -22,7 +22,7 @@ class RepVGGBlock(layers.Layer):
|
|
22 |
groups=1, deploy=False, use_se=False, **kwargs):
|
23 |
super(RepVGGBlock, self).__init__(**kwargs)
|
24 |
self.config_initial_in_channels = in_channels
|
25 |
-
self.config_out_channels = out_channels
|
26 |
self.config_kernel_size = kernel_size
|
27 |
self.config_strides_val = stride
|
28 |
self.config_groups = groups
|
|
|
22 |
groups=1, deploy=False, use_se=False, **kwargs):
|
23 |
super(RepVGGBlock, self).__init__(**kwargs)
|
24 |
self.config_initial_in_channels = in_channels
|
25 |
+
self.config_out_channels = out_channels # Corrected: Removed extra '_out'
|
26 |
self.config_kernel_size = kernel_size
|
27 |
self.config_strides_val = stride
|
28 |
self.config_groups = groups
|