Wangrf:

File “D:\NTIRE\HRNet\network_code1.py”, line 87, in forward
x2 = torch.cat((x2, x3), 1) # out: batch * (128 + 64) * 64 * 64
RuntimeError: Sizes of tensors must match except in dimension 2. Got 36 and 37 (The offending index is 0)


check shapes of x2 and x3. Error says x2.shape[0] = 36, x3.shape[0] = 37, but you can have different shapes only at dimension you are concatenating along.