deepke/example/ae/standard/conf/model/gcn.yaml

8 lines
135 B
YAML

model_name: gcn
num_layers: 3
input_size: ??? # 使用 embedding 输出的结果,不需要指定
hidden_size: 100
dropout: 0.3