https://github.com/ARCAFF/ARCCnet/blob/8f1afab910ec423992d5edbede05d7a802bf34f5/arccnet/models/cutouts/mcintosh/train_utils.py#L237 maybe validate or something?