This code is for the following paper:

H. He, C. Wen, S. Jin, and G. Y. Li, “Model-driven deep learning for MIMO detection,” IEEE Trans. Signal Process., vol. 68, pp. 1702–1715, Feb. 2020.

Please cite the paper when use this code. weight_mat should be revised as your path to store the trainable variables.

Questions/suggestions/comments about OAMP-Net: hehengtao@seu.edu.cn