fix: add check for NaN validation loss in EarlyStopping#28

Bjarten merged 2 commits intomainBjarten/early-stopping-pytorch:mainfrom

fix/early-stopping-nan-checkBjarten/early-stopping-pytorch:fix/early-stopping-nan-checkCopy head branch name to clipboard