quanweikikai

you mean GPU memory? you can modify batch size to control memory consuming. Usually, this one will not take that much memory in my case.

The code did not run on my PC but I customized it and add the TEST phase to the network so it work well. If you need I can give...