tdm 单机预测评估时卡住没反应

[ideal-lee]

按照教程单机版demo，在完成训练和聚类后，进行预测和评估时，运行一会儿算loss时卡住了，如图这样，后面就一直卡住没反应，直到连接集群时间过长退出，请问是什么原因？使用的就是user_behavier数据集
命令是python train.py --run_mode=local --config=config.predict.json

[ideal-lee]

@zhuhan1236 请教各位大佬

[Jim59-Chen]

你好，你参考教程完成了TDM的训练吗？
我在这一步遇到点问题，想请教一下您。
我这边用虚拟机跑train.py失败了，你是使用GPU来跑这个程序吗