Reply To: Memory Error 3 in modified UNet

#57864
Youssef MENJOURYoussef MENJOUR
Admin
    @youssefmenjour

    Hello Peter,

    Take a closer look, my partial solution has a different architecture.

    On the current problematic architecture the A and B blocks are in parallel.  That’s the problem, the version of HAIBAL you have does not handle this parallelism properly (Fixed and available in next release)

    peter

    To solve this problem I propose a small modification that will avoid the error.
    You have to set the architecture in sequence as follows.

    partial solution

    This time we are in sequential, Block C, D and E follow each other and erase the bug at the same time.
    This solution is partial because if we want to be optimized in terms of speed we will have to run C and D in parallel (this will allow us to make predictions while training the model in an optimized way).