metricsubs-segmenter / model_consts.py
metricv's picture
Update model
d06f65e verified
raw
history blame
231 Bytes
if __package__ == None or __package__ == "":
from utils import get_upenn_tags_dict
else:
from .utils import get_upenn_tags_dict
input_size = len(get_upenn_tags_dict())
embedding_size = 256
hidden_size = 256
num_layers = 2