CNTK 分布式崩溃 - 测试版 7



我正在运行CIFAR 10的变体,以利用我的数据。

我收到以下错误:

Traceback (most recent call last):
  File "CNTK_Train.py", line 158, in <module>
    checkpoint_path = "C:/projects/RoboLabs/CognitiveServices/ML_Models/DocSuite/Doc_Classify/checkpoints/CNTK_VGG9")
  File "CNTK_Train.py", line 80, in train_and_evaluate
    trainer.save_checkpoint(os.path.join(checkpoint_path + "_{}.dnn".format(current_epoch)))
  File "C:localAnaconda3-4.1.1-Windows-x86_64envscntk-py34libsite-packagescntktrainer.py", line 138, in save_checkpoint
    super(Trainer, self).save_checkpoint(filename, _py_dict_to_cntk_dict(external_state))
  File "C:localAnaconda3-4.1.1-Windows-x86_64envscntk-py34libsite-packagescntkcntk_py.py", line 1774, in save_checkpoint
    return _cntk_py.Trainer_save_checkpoint(self, *args)
RuntimeError: Runtime exception

我用于带有检查点的训练循环的代码在这里:

while updated:
    data=train_reader.next_minibatch(minibatch_size, input_map=input_map) # fetch minibatch.
    updated=trainer.train_minibatch(data)                                 # update model with it
    progress_printer.update_with_trainer(trainer, with_metric=True)       # log progress
    epoch_index = int(trainer.total_number_of_samples_seen/epoch_size)
    if current_epoch != epoch_index:                                      # new epoch reached
        progress_printer.epoch_summary(with_metric=True)
        current_epoch=epoch_index            
    if current_epoch % 25 == 0:
        trainer.save_checkpoint(os.path.join(checkpoint_path + "_{}.dnn".format(current_epoch)))

欢迎提供见解。 我正在积极调试。

您是否可能以 Linux 样式指定的 Windows 环境路径格式运行。在Windows上,路径应该是类似于'X:\Repos\CNTK\Examples\Image\Classification\ResNet\Python\Models\resnet20_0.dnn'。建议您在传递给save_model的路径字符串中尝试 os.path.join 而不是硬编码/或 \。

这似乎在最新版本中得到了解决,该版本以不同的方式执行检查点。 解决方案是升级 CNTK 版本。 使用从版本 9 开始的会话 API。

最新更新