复现过程报错

[rorozi]

真诚请教：
想要复现论文的实验，我把作者readme中vgg和content_net两个模型的文件下载下来新建了文件夹按照说的路径存放，然后把RHSEDB数据集放在了dataset路径中，然后去运行train_SDNet.py，main_train.py会报段错误：
Segmentation fault (core dumped)
求助作者这是什么原因呀？

[rorozi]

运行load_data_for_SDNet.py会报这样的错误：
style_single_image = sample_batched['style_single_image'][0].numpy()
KeyError: 'style_single_image'

[MengLi-l1]

运行load_data_for_SDNet.py会报这样的错误：
style_single_image = sample_batched['style_single_image'][0].numpy()
KeyError: 'style_single_image'

这个文件是数据加载模型，你直接运行这个文件会执行文件‘if name == 'main':’下面的代码，这个代码在正常训练并不会用到，只是我简单测试数据集加载效果的，sample_batched中的key值是不对的，

[MengLi-l1]

真诚请教：
想要复现论文的实验，我把作者readme中vgg和content_net两个模型的文件下载下来新建了文件夹按照说的路径存放，然后把RHSEDB数据集放在了dataset路径中，然后去运行train_SDNet.py，main_train.py会报段错误：
Segmentation fault (core dumped)
求助作者这是什么原因呀？

至于这个问题应该不是我代码导致的，你重新检查下运行环境试试看，或者重新配置下环境。train_SDNet.py单独训练SDNet，main_train.py则顺序训练全部三个网络