Added: cfg.resolution = resolution at line 240 to prevent the error: The size of tensor a (64) must match the size of tensor b (96) at non-singleton dimension 4"
Added: cfg.resolution = resolution at line 240 to prevent the error: The size of tensor a (64) must match the size of tensor b (96) at non-singleton dimension 4"