File size: 775 Bytes
2571cc4 |
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}")
|