Train one model for 6 specimen

[legendjack]

Hi Vivek,

Thank you for your work.

I ran your code on a GPU. I wonder if I run this command cd experiments/deepfluoro & srun python train.py, will it train 6 models for 6 specimen. So how to train a single model for 6 specimen.

And I ran evaluate.py, the model got a rough registration result. In register.py, it performed a gradient descent optimization after model prediction, and the result is much more precise. Is it possiable to train a model to get a precise result directly?

I am a beginner in deep learning registration, look forward to your reply. Thank you.