src/modules/losses.py

"All criterion functions."
from torch.nn import MSELoss, CrossEntropyLoss
from src.utils.mapper import configmapper

configmapper.map("losses", "mse")(MSELoss)
configmapper.map("losses", "CrossEntropyLoss")(CrossEntropyLoss)