File size: 225 Bytes
cc9c7ee | 1 2 3 4 5 6 7 | "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)
|