use length of saved_logit array when evaluating logits

[nickdnickd]

I ran into this on the first run similar to issue https://github.com/jannisborn/covid19_ultrasound/issues/103 When I only had one fold by default. This should hopefully allow testing of any number of folds.

[jannisborn]

Thanks again, makes perfect sense!