Skip to content

Commit 2370243

Browse files
authored
fix padding not defined in cifar10 models
1 parent 833fae5 commit 2370243

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

cifar/models/simplenet.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -297,6 +297,7 @@ def _make_layers(self, scale: float):
297297
# check to convert any possible integer value to its decimal counterpart.
298298
custom_dropout = None if custom_dropout is None else float(custom_dropout)
299299
kernel_size = 3
300+
padding = 1
300301
if layer_type == ['k1']:
301302
kernel_size = 1
302303
padding = 0

0 commit comments

Comments
 (0)