micmic123 / QmapCompression

Official implementation of "Variable-Rate Deep Image Compression through Spatially-Adaptive Feature Transform", ICCV 2021
81 stars 16 forks source link

error:the size of tensor a(512) must match the size of tensor b(500) at non-singleton dimension 3 #6

Open linshiyan opened 1 year ago

linshiyan commented 1 year ago

Using a target size (torch.Size([1,3,0,500)) that is different to the input size (torch.Size([1, 3,512, 512])).

error:the size of tensor a(512) must match the size of tensor b(500) at non-singleton dimension 3

How to solve this problem?