Update utils.py

PlayVoice · Sep 1, 2023 · deff80f · deff80f
1 parent 4fbbe63
commit deff80f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/grad_extend/utils.py b/grad_extend/utils.py
@@ -71,3 +71,7 @@ def save_plot(tensor, savepath):
     plt.savefig(savepath)
     plt.close()
     return
+
+
+def print_error(info):
+    print(f"\033[31m {info} \033[0m")