I am trying to create trace file. Related code part is here:
def center_crop(self, layer, target_size): _, _, layer_height, layer_width = layer.size() diff_y = (layer_height - target_size) // 2 diff_x = (layer_width - target_size) // 2 return layer[:, :, diff_y:(diff_y + target_size), diff_x:(diff_x + target_size)] def forward(self, x, bridge): up = self.up(x) crop1 = self.center_crop(bridge, up.shape[2:]) out = torch.cat([up, crop1], 1) out = self.conv_block(out)
It gives the warning at this line:
return layer[:, :, diff_y:(diff_y + target_size), diff_x:(diff_x + target_size)]
Then, I loaded this trace file in C++ and made prediction for a test image. Then I compared the result with the output in python. I observed that results are very different. I guess that the reason originates from that warning.