mirror of https://github.com/hpcaitech/ColossalAI
14 lines
220 B
Python
14 lines
220 B
Python
|
#!/usr/bin/env python
|
||
|
|
# -*- encoding: utf-8 -*-
|
||
|
|
|
||
|
|
from abc import ABC, abstractmethod
|
||
|
|
|
||
|
|
|
||
|
|
class BaseLoss(ABC):
|
||
|
|
"""Absctract loss class
|
||
|
|
"""
|
||
|
|
|
||
|
|
@abstractmethod
|
||
|
|
def calc_loss(self, *args, **kwargs):
|
||
|
|
pass
|