Dicece check for float target (Project-MONAI#5326)

Fixes Project-MONAI/tutorials#987 ### Types of changes <!--- Put an `x` in all the boxes that apply, and remove the not applicable items --> - [x] Non-breaking change (fix or new feature that would not break existing functionality). - [ ] Breaking change (fix or new feature that would cause existing functionality to change). - [ ] New tests added to cover the changes. - [ ] Integration tests passed locally by running `./runtests.sh -f -u --net --coverage`. - [ ] Quick tests passed locally by running `./runtests.sh --quick --unittests --disttests`. - [ ] In-line docstrings updated. - [ ] Documentation updated, tested `make html` command in the `docs/` folder. Signed-off-by: myron <amyronenko@nvidia.com>

Author: myron

.github/workflows/pythonapp-gpu.yml

@@ -16,7 +16,7 @@ concurrency:
 
 jobs:
  GPU-quick-py3: # GPU with full dependencies
- if: github.repository == 'Project-MONAI/MONAI'
+ if: ${{ github.repository == 'Project-MONAI/MONAI' && github.event.pull_request.merged != true }}
  strategy:
  matrix:
  environment:
@@ -124,6 +124,7 @@ jobs:
  python -m pip install -r requirements-dev.txt
  python -m pip list
  - name: Run quick tests (GPU)
+ if: github.event.pull_request.merged != true
  run: |
  git clone --depth 1 \
  https://github.com/Project-MONAI/MONAI-extra-test-data.git /MONAI-extra-test-data

monai/losses/dice.py

@@ -711,8 +711,9 @@ def ce(self, input: torch.Tensor, target: torch.Tensor):
  "Using argmax (as a workaround) to convert target to a single channel."
  )
  target = torch.argmax(target, dim=1)
- else: # target has the same shape as input, class probabilities in [0, 1], as floats
- target = target.to(input) # check its values are in [0, 1]??
+ elif not torch.is_floating_point(target):
+ target = target.to(dtype=input.dtype)
+
  return self.cross_entropy(input, target)
 
  def forward(self, input: torch.Tensor, target: torch.Tensor) -> torch.Tensor: