The model is working, but the result is not very good. Should probably train without the empty patches...
-
weights after 10 iterations: weights-UNETR_2D-src-.-bce-nf-32-bs-6-AdamW-oneCycle_001.weights.h5
-
weights after 38 iterations: weights-UNETR_2D-src-.-bce-nf-32-bs-6-AdamW-oneCycle.weights.h5
-
unetr.-.2025-02-05.17.58.26.090084.keras corresponds to the weights after 38 iterations