ibrahim comments

Repositories
Issues
Comments

Results 2 comments of


                                            ibrahim

Segformer/Transformer Backbone

You can give [Mix Visual Transformer](https://smp.readthedocs.io/en/latest/encoders.html#:~:text=20M-,Mix%20Visual%20Transformer,-%23) a try. It's a [SegFormer](https://huggingface.co/nvidia/mit-b0) encoder.

DiceLoss when multiclass mode throwing an assertion error

> @qubvel thanks for the reply. > > So, if we're to use `multiclass` mode in this `DiceLoss` function, `y_true` should be in `B, H, W` shape and `y_pred` should...