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