@@ -502,7 +502,8 @@ def main():
502
502
f'{ pixelcount } ) does not match the '
503
503
f'sum of all pixels ({ pixels } ) in the layer\' s `in_sequences`.' )
504
504
input_dim [ll ] = conf_input_dim [ll ]
505
- if operator [ll ] != op .CONV1D :
505
+
506
+ if not (operator [ll ] == op .CONV1D or (operator [ll ] == op .NONE and input_dim [ll ][1 ] == 1 )):
506
507
if pool_stride [ll ][0 ] != pool_stride [ll ][1 ]:
507
508
eprint (f'{ layer_pfx (ll )} { op .string (operator [ll ])} does not support '
508
509
f'non-square pooling stride (currently set to '
@@ -512,6 +513,7 @@ def main():
512
513
(input_dim [ll ][1 ] + pool_stride [ll ][1 ] - pool [ll ][1 ]
513
514
- pool_dilation [ll ][1 ] + 1 ) // pool_stride [ll ][1 ]]
514
515
else :
516
+ pool [ll ][1 ] = 1
515
517
pooled_size = [(input_dim [ll ][0 ] + pool_stride [ll ][0 ] - pool [ll ][0 ]
516
518
- pool_dilation [ll ][0 ] + 1 ) // pool_stride [ll ][0 ],
517
519
1 ]
@@ -521,7 +523,7 @@ def main():
521
523
eprint (f'{ layer_pfx (ll )} Pooling or zero-padding results in a zero data '
522
524
f'dimension (input { input_dim [ll ]} , result { pooled_dim [ll ]} ).' )
523
525
524
- if operator [ll ] != op .CONV1D :
526
+ if not ( operator [ll ] == op .CONV1D or ( operator [ ll ] == op . NONE and input_dim [ ll ][ 1 ] == 1 )) :
525
527
if stride [ll ][0 ] != stride [ll ][1 ]:
526
528
eprint (f'{ layer_pfx (ll )} { op .string (operator [ll ])} does not support '
527
529
f'non-square stride (currently set to { stride [ll ][0 ]} x{ stride [ll ][1 ]} ).' )
0 commit comments