mirror of
https://github.com/tencentmusic/cube-studio.git
synced 2024-12-21 06:19:31 +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
|