Hello, is it okay for me to train a liver dataset with this code? I lose drops when training but DICE barely when testing