Skip to content

Conversation

@BoyuShen2004
Copy link
Contributor

  • Add data type conversion (_prepare_images/_prepare_labels) to prevent uint8/float16 mismatch
  • Implement CellMapBalancedLoss with comprehensive NaN/inf handling for sparse segmentation
  • Add batch skipping for empty batches to prevent NaN contamination in progress bars
  • Replace problematic pos_weight computation with uniform weights to avoid numerical instability
  • Add limit_train_batches/limit_val_batches to prevent infinite epochs from large datasets
  • Enhance loss function with gradient-connected safety values and extensive numerical stability checks

- Add data type conversion (_prepare_images/_prepare_labels) to prevent uint8/float16 mismatch
- Implement CellMapBalancedLoss with comprehensive NaN/inf handling for sparse segmentation
- Add batch skipping for empty batches to prevent NaN contamination in progress bars
- Replace problematic pos_weight computation with uniform weights to avoid numerical instability
- Add limit_train_batches/limit_val_batches to prevent infinite epochs from large datasets
- Enhance loss function with gradient-connected safety values and extensive numerical stability checks
@BoyuShen2004 BoyuShen2004 deleted the fix-cellmap-training-issues branch December 8, 2025 06:34
@BoyuShen2004 BoyuShen2004 restored the fix-cellmap-training-issues branch December 8, 2025 06:35
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant