连接modelarts-开发环境-Notebook程序卡住,同样代码笔记本cpu没卡
【截图信息】
日志提示数据侧获取不到数据,有可能数据处理时间较长,请查看数据处理环节是否有性能不够好,导致数据处理时间过长,数据测等候超时导致
可以单独遍历下数据 看看遍历出每个step(batch)的耗时,对这部分进行相关优化
耗时查看方法,可以仿照一下例子看下
time_start = time.time()
for item in dataset.create_tuple_iterator(num_epochs=1, output_numpy=True):
time_end = time.time()
print("step time:", time_end - time_start)
time_start = time_end