File size: 781 Bytes
204da06 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
from .normalization import NORM2CLS
from .embeddings import EBB2CLS
from .activation import ACT2CLS
from .attention import ATTN2CLS
from .mlp import MLP2CLS
from .layers import LAYER2CLS
def print_layer_options():
print("Activation options:")
for option in ACT2CLS:
print(f" {option}")
print("\nAttention Layer options:")
for option in ATTN2CLS:
print(f" {option}")
print("\nEmbedding Layer options:")
for option in EBB2CLS:
print(f" {option}")
print("\nBert Layer options:")
for option in LAYER2CLS:
print(f" {option}")
print("\nMLP Layer options:")
for option in MLP2CLS:
print(f" {option}")
print("\nNormalization options:")
for option in NORM2CLS:
print(f" {option}")
|