Commit be48e402 authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu 🖥
Browse files

created file with multiple potential loss classes

parent ef285c19
"""Deep Learning Loss Functions
Description:
-------------
This file contains several loss functions which, could be useful when training the BrainMapper Network.
Usage
-------------
To use content from this folder, import the functions and instantiate them as you wish to use them:
from utils.losses import LossFunction as loss_functions
loss = loss_functions.L1Loss()
"""
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.nn.modules.loss import _Loss, _WeightedLoss
class L1Loss():
pass
class DiceLoss():
pass
class MSELoss():
pass
class CrossEntropyLoss():
pass
class CrossDiceLoss():
pass
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment