配置文件内容如下:
config = {
'input_features': [
{
'name': 'image_path',
'type': 'image',
'preprocessing': {'num_processes': 4},
'encoder': 'stacked_cnn',
'conv_layers': [
{'num_filters': 32, 'filter_size': 3, 'pool_size': 2, 'pool_stride': 2},
{'num_filters': 64, 'filter_size': 3, 'pool_size': 2, 'pool_stride': 2, 'dropout': 0.4}
],
'fc_layers': [{'output_size': 128, 'dropout': 0.4}]
}
],
'output_features': [{'name': 'label', 'type': 'binary'}],
'trainer': {'epochs': 5}
}
4条答案
按热度按时间j8yoct9x1#
你好@AnaMiguelRodrigues1,感谢你提出这个问题和建议的解决方法!
我只是想了解一些事情:
wixjitnu2#
我正在使用最新版本,ludwig==0.8.dev0。错误是语法错误,只是在配置中'encoder'字段没有正确定义。
rmbxnbpk3#
明白了!您是否愿意将修复贡献给Ludwig?
xriantvc4#
我很想纠正它,但在Ludwig的官方网站上(https://ludwig.ai/latest/examples/mnist/),我没有权限更改网站和纠正配置。