Deal with missing parts?

[ethanygao]

Hi, does the missing parts are ignored during training by weighted_loss (the weighted bce)? But it seems this is still work in progress, so does it mean the current model cannot be trained on data with missing parts?

Thanks! @wbenbihi