update, with 25 epochs 3 repetitions and the TE learning rate at the same learning rate as the unet,
update, with 25 epochs 3 repetitions and the TE learning rate at the same learning rate as the unet, the bleeding I think is fixed, I'm no so sure yet because is undertrained, I will resume the training for 25 more ephocs but the results are promising
