Raises warning (pytorch 1.6),

[saruarlive]

Dear @migonch ,

The CRF class raises warning!
Could you kindly provide guidance on how to fix that? I modified the code a little, but still raises the warning.
UserWarning: Mixed memory format inputs detected while calling the operator. The operator will output contiguous tensor even if some of the inputs are in channels_last format. (Triggered internally at /opt/conda/conda-bld/pytorch_1595629427478/work/aten/src/ATen/native/TensorIterator.cpp:918.)

[mishgon]

Dear @saruarlive ,

unfortunately, this implementation of CRF layer does not support the channels_last format of input tensor. Make sure that input tensor has size (N, C, H, W, ...), where N is batch size, C is number of segmentation classes, and (H, W, ...) is spatial size.