My teacher asked me, a freshman, to finish this terrible work. It’s obviously not a work for a common undergraduate freshman.

Here is the website of the project: https://github.com/MahdiShAhmadi/ReDMark
and as for the torch version, I use torch2.3.0+cu121

I tried to deal with it by torch, but I failed, please help me☠️