Commit 2370243

authored

fix padding not defined in cifar10 models

1 parent 833fae5 commit 2370243Copy full SHA for 2370243

File tree

-0

lines changed

-0

lines changed

Lines changed: 1 addition & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -297,6 +297,7 @@ def _make_layers(self, scale: float):`
`297`	`297`	`# check to convert any possible integer value to its decimal counterpart.`
`298`	`298`	`custom_dropout = None if custom_dropout is None else float(custom_dropout)`
`299`	`299`	`kernel_size = 3`
	`300`	`+ padding = 1`
`300`	`301`	`if layer_type == ['k1']:`
`301`	`302`	`kernel_size = 1`
`302`	`303`	`padding = 0`

Comments

(0)