14 lines
408 B
YAML
14 lines
408 B
YAML
model_name: cnn
|
||
|
||
in_channels: ??? # 使用 embedding 输出的结果，不需要指定
|
||
out_channels: 100
|
||
kernel_sizes: [3, 5, 7] # 必须为奇数，为了保证cnn的输出不改变句子长度
|
||
activation: 'gelu' # [relu, lrelu, prelu, selu, celu, gelu, sigmoid, tanh]
|
||
pooling_strategy: 'max' # [max, avg, cls]
|
||
keep_length: True
|
||
dropout: 0.3
|
||
|
||
# pcnn
|
||
use_pcnn: False
|
||
intermediate: 80
|