72f89ffbeaad419cad9abc7c2d6a68bc1304396c,torchsample/modules/_utils.py,,validate_loss_input,#Any#,22

Before Change

            loss_fn = eval("F.%s" % loss)
        except:
            raise ValueError("Invalid loss string input - must match torch function exactly!")
        return loss_fn
    elif callable(loss):
        return loss
    else:
        raise ValueError("Invalid metric input")

After Change

        raise ValueError("Invalid metric input")

def validate_loss_input(loss):
    dir_f = dir(F)
    loss_fns = [d.lower() for d in dir_f]
    if isinstance(loss, str):
        try:
            str_idx = loss_fns.index(loss.lower())
        except:
            raise ValueError("Invalid loss string input - must match pytorch function.")
        return getattr(F, dir(F)[str_idx])
    elif callable(loss):

Instances