split_with_sizes expects split_sizes to sum exactly to 128 (input tensor's size at dimension 1), but got split_sizes=[42, 42, 42]

[dengfenglai]

输入
input = torch.randn(1, 3, 320, 180)
报错如题 。
请问怎么修改

网络配置：

scale: 4

rgb_range: 255

colors: 3

m_elan: 24

c_elan: 60

n_share: 1

r_expand: 2

act_type: 'relu'

window_sizes: [4, 8, 16]