Overflowu7

Results 2 issues of Overflowu7

I want the weight_ce=1, weight_dice=1 to decrease from 1 to 0 as the epoch increases, but if I add the epcoh parameter to the _build_loss function, there are other coupling...

When a finish the train epcoh and do the validation ,the "CUDA out of memory" appeared. I tried to use two Gpus at the same time but it didn't seem...