问题:
mindinsight 在modelart上使用报错问题。
在添加summary后,正常训练一些epoch后,运行报错:
解决方案:
使用SummaryCollector的时候,需要将代码块放到 if __name__ == __main__:
里面。
MindSpore官方教程已更新,可以参考最新教程的写法:收集Summary数据 — MindSpore master 文档
类似这样的代码:
def train():
summary_collector = SummaryCollector(summary_dir='./summary_dir')
...
model.train(...., callback=[summary_collector])
if __name__ == '__main__':
train()