gen_rnet_data

[Sarah-Elfanty]

Hi, I've been trying to train this model from scratch on a local dataset, but I've been encountering a CUDA memory issue when i run gen_Rnet_train_data.py
I'm using a 1050ti , 4gb GPU with 16 gb RAM.

[Sierkinhane]

Maybe you should reduce the batch size of testimageloader

[liaomuqin]

我也遇到了同样的问题，请问怎么解决。。。