I have been struggling for re-implementing this thing using pytorch for a long time, yet still no break through. Does anybody have any suggestions?