Training error

[Duranin]

Hello, I downloaded the corresponding pre-trained model "mobilevitv2-1.0.pt" from [ml-cvnets] and configured the relevant parameters. However, during the training process, there are the following issues:
"SerWarning: An output with one or more elements was resized since it had shape [8388608], which does not match the required output shape [1, 128, 256, 256]."“
This warning covers almost every layer. How should I correct it? Thank you for your support and work.

[goutamyg]

Hi,

Thank you for your interest in our work. Regarding the warning you encountered, it seems to be related to an older version of PyTorch, which has been addressed in the latest release. Since I conduct model training on a remote server, I haven't attempted to resolve or suppress this warning (although I acknowledge that it is annoying). Rest assured, despite the warning message, the training process remains unaffected.

To verify the training and validation loss, as well as IoU values, you can refer to the log file located in the output/logs/ directory.