Solved by Research_ZZ in post #3 
              
              
                finally tried auto encoder with FNN and CNN,  use MSELoss and adjusted weight decay, and learning rate, seems to work. Loss decreasing, and can restore image reasonable well.