训练时遇到的问题

[yimoo4J]

您好，非常感谢您将代码开源。 我现在遇到的问题是在训练时程序会卡住不动

我的训练数据是从您的百度网盘下载后直接解压的，结构如下

data ----- kaist -----------kaist_train ----------------------set05 ----------------------------- V000 -------------------------------------visble -------------------------------------lwir --------------------- set04 ---------------------- ...... 运行test.py时正常。 请问您知道问题出在什么地方吗？

[CalayZhou]

您好，如果test.py没问题的话，可以检查一下GPU显存是否占用，这个问题之前没有遇到过，应该与环境配置有关，可以调试看看是在哪一行卡住了

[NNNNerd]

你好 @yimoo4J 请问你的问题解决了吗？