bayesiancnn/layers/__init__.py

6 lines
262 B
Python
Raw Normal View History

from .BBB.BBBConv import BBBConv2d as BBB_Conv2d
from .BBB.BBBLinear import BBBLinear as BBB_Linear
from .BBB_LRT.BBBConv import BBBConv2d as BBB_LRT_Conv2d
from .BBB_LRT.BBBLinear import BBBLinear as BBB_LRT_Linear
from .misc import FlattenLayer, ModuleWrapper