I'd like to ask where the test code is located? And where can I use the trained model to test the test set data to obtain metrics such as DICE?