rusheniii

Results 2 issues of rusheniii

**Is your feature request related to a problem? Please describe.** Sometimes the loss jumps around after a large number of epochs during training. It would be nice to use a...

**Is your feature request related to a problem? Please describe.** When you are training an LDDMM atlas, if you don't set the parameters correctly, the loss can go to Nan....