import torch.nn as nn


class Loss_Fn:
    def cross_entropy():
        return nn.CrossEntropyLoss()

import torch.nn as nn


class Loss_Fn:
    def cross_entropy():
        return nn.CrossEntropyLoss()
