CCNet
CCNet copied to clipboard
about INF
excuse me . Is there an another method? the torch.diag methon isn't supported by onnx version and when i use torch.eye ,my tensorrt doesn‘t support it either.
@Thatboy7 You could drop the INF if you train the model again. The overlapping pixels have a minor effect on the final performance.