Eloy Roura
Results
3
comments of
Eloy Roura
Would be great to have this merged :)
I had a similar issue, first I modified the serilization.py class to hardcoding the cuda device I wanted to use. Now, the workarround I found was to add map_location={'cuda:1': 'cuda:0'}...
@jocicmarko Thanks for your code, it helped me quite a lot to get my unet architecture. @Kaido0 I run into the same issue with the dice computation and this is...