Your keras model defines 10
filters with kernel_size=9
in the first conv layer, while in your PyTorch model you define 192
filters with kernel_size=10
. Keras’ filters
is equal to out_channels
.
EDIT: Also it seems to me that your Keras input hat 76
channels. You should transpose the input to get similar results to [batch, channels, length]
.