From 10ed90dddeaa03daef9141228adc597a028a85e0 Mon Sep 17 00:00:00 2001
From: FenghaoZhu <61408487+FenghaoZhu@users.noreply.github.com>
Date: Sat, 26 Oct 2024 16:02:59 +0800
Subject: [PATCH] Update README.md

---
 README.md | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index f9cbd10..1fef127 100644
--- a/README.md
+++ b/README.md
@@ -1,2 +1,8 @@
 # Matrix-Inverse-Free-WMMSE
-This is the code implementation for the Matrix-Inverse-Free  MU-MIMO WMMSE Beamforming Algorithm.
+This is the code implementation for the [Matrix-Inverse-Free MU-MIMO WMMSE Beamforming Algorithm](https://ieeexplore.ieee.org/abstract/document/10021866).
+This is a pytorch version that you can easily run with "python inverse_free.py". 
+
+## Files in this repo
+`inverse_free.py`: The main function. Can be directly run to get the results.
+
+`utils.py`: This file contains the util functions.