Running python train_multi_lcfed.py, I am getting this kind of result
WARNING:root:NaN or Inf found in input tensor.
[Test] epoch 0 testing Site 2 [2025-10-12 14:29:37,768]
INFO:logger:[Test] epoch 0 testing Site 2
WARNING:root:NaN or Inf found in input tensor.
[Test] epoch 0 testing Site 3 [2025-10-12 14:29:37,769]
INFO:logger:[Test] epoch 0 testing Site 3
WARNING:root:NaN or Inf found in input tensor.
[Test] epoch 0 testing Site 4 [2025-10-12 14:29:37,769]
INFO:logger:[Test] epoch 0 testing Site 4
WARNING:root:NaN or Inf found in input tensor.
WARNING:root:NaN or Inf found in input tensor.
[INFO] Dice Overall: nan Best Dice 0.00
At the end of the epoch, it also gives Dice Overall: nan Best Dice 0.00. Can u please tell me how to solve it?
Running python train_multi_lcfed.py, I am getting this kind of result
WARNING:root:NaN or Inf found in input tensor.
[Test] epoch 0 testing Site 2 [2025-10-12 14:29:37,768]
INFO:logger:[Test] epoch 0 testing Site 2
WARNING:root:NaN or Inf found in input tensor.
[Test] epoch 0 testing Site 3 [2025-10-12 14:29:37,769]
INFO:logger:[Test] epoch 0 testing Site 3
WARNING:root:NaN or Inf found in input tensor.
[Test] epoch 0 testing Site 4 [2025-10-12 14:29:37,769]
INFO:logger:[Test] epoch 0 testing Site 4
WARNING:root:NaN or Inf found in input tensor.
WARNING:root:NaN or Inf found in input tensor.
[INFO] Dice Overall: nan Best Dice 0.00
At the end of the epoch, it also gives Dice Overall: nan Best Dice 0.00. Can u please tell me how to solve it?