mirror of
https://github.com/tencentmusic/cube-studio.git
synced 2025-03-07 15:08:51 +08:00
13 lines
299 B
Python
13 lines
299 B
Python
class TrainingConfig(object):
|
|
epochs = 10
|
|
batch_size = 16 # batch_size != 1
|
|
lr = 0.0005
|
|
|
|
class BiLSTMConfig(object):
|
|
input_size = 768 # embedding size
|
|
hidden_size = 512
|
|
|
|
class BiLSTMCRFTrainConfig(object):
|
|
epochs = 10
|
|
batch_size = 16 # batch_size != 1
|
|
lr = 0.0005 |