Skip to content

Commit 3644674

Browse files
authored
Feature/blocks (#48)
*implemented standard block *now BatchNorm and not layernorm *allow for either pre- or post normalisation
1 parent 6cda667 commit 3644674

File tree

9 files changed

+71
-379
lines changed

9 files changed

+71
-379
lines changed

src/linax/blocks/__init__.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,11 @@
11
"""This module contains the blocks implemented in Linax."""
22

33
from linax.blocks.base import Block, BlockConfig
4-
from linax.blocks.linoss import LinOSSBlock, LinOSSBlockConfig
5-
from linax.blocks.lru import LRUBlock, LRUBlockConfig
6-
from linax.blocks.s5 import S5Block, S5BlockConfig
4+
from linax.blocks.standard import StandardBlock, StandardBlockConfig
75

86
__all__ = [
97
"BlockConfig",
108
"Block",
11-
"LinOSSBlockConfig",
12-
"LinOSSBlock",
13-
"LRUBlockConfig",
14-
"LRUBlock",
15-
"S5BlockConfig",
16-
"S5Block",
9+
"StandardBlockConfig",
10+
"StandardBlock",
1711
]

src/linax/blocks/linoss.py

Lines changed: 0 additions & 143 deletions
This file was deleted.

src/linax/blocks/lru.py

Lines changed: 0 additions & 152 deletions
This file was deleted.

0 commit comments

Comments
 (0)