Refer to this 639 answer and calculate the padding needed for conv2D transpose of pytorch. It’ll enable you to convert between tflearn.layers.conv.conv_2d_transpose with non-unit stride to pytorch version of conv2D