The code doesn't work

[Cherrypower]

Hello GTS author, I just started running the source code and got an error message:
(UserWarning: masked_ fill_ received a mask with dtype torch. uint8, this behavior is now deprecated, please use a mask with dtype torch. bool instead.)
But when I replace it, the results will only be displayed to fold: 1, epoch: 1. There is no result after that. Find the solution, Thank you!

[ShichaoSun]

This is warning, and I think maybe not care about it.

And may use xxx.bool() to convert it to bool variable.
Help it can help you.

BTW, PyTorch has updated too fast, and this code is implemented by PyTorch 0.4.1 in 2019, which is too out-of-date. If some kpi may not work, you can degrade your PyTorch version.