I need to call backward function for two different models, separately.
# Loss_1 #---------------------- Losses_1 = [] with autograd.record(): for