训练时遇到的问题

[yimoo4J]

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

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

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

[CalayZhou]

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

[NNNNerd]

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