search

gmalivenko / pytorch2keras

PyTorch to Keras model convertor
https://pytorch2keras.readthedocs.io/en/latest/
MIT License
858 stars 143 forks source link

keras model's summary , first output shape is [(None, 1, 28,28)] #104

Open sunpengfei1122 opened 4 years ago

sunpengfei1122 commented 4 years ago

Layer (type) Output Shape Param #

input_0 (InputLayer) [(None, 28, 28, 1)] 0

27_pad (ZeroPadding2D) (None, 30, 30, 1) 0

the format is [()] , is it for the multi-input? and how can I change it into (). for example, change [(None,1,28,28)] to (None, 1, 28, 28). Thanks