mirror of https://github.com/hpcaitech/ColossalAI
7 lines
263 B
Python
7 lines
263 B
Python
|
from .base_loss import BaseLoss
|
||
|
|
from .cross_entropy_2d import CrossEntropyLoss2D
|
||
|
|
from .cross_entropy_2p5d import CrossEntropyLoss2p5D
|
||
|
|
from .cross_entropy_3d import CrossEntropyLoss3D
|
||
|
|
|
||
|
|
__all__ = ['CrossEntropyLoss2D', 'CrossEntropyLoss2p5D', 'CrossEntropyLoss3D']
|