Thanks. I will check this.
Can I give you another question?
Suppose that both forward and backward function of autograd are implemented correctly, What are the differences between two custom loss functions that are based on respectively ‘autograd’ and ‘nn.Module’?