devavratTomar / SST

Self-Supervised Generative Style Transfer for One-Shot Medical Image Segmentation
27 stars 7 forks source link

Questions about the third step of the training #4

Open AXu0511 opened 2 years ago

AXu0511 commented 2 years ago

Following your training steps, after running the first step, the Dice value of my registration model(UnetFlowModel) was 0.768. Then I run the second step. When I run the third step, line 338 of the code is the test code, and the test result showed that the performance of the alignment model dropped to 0.73. Should I use the pre-trained StyleEncoder when running the third step, and if not, what is the purpose of the second step?

1668922404559