After reading your code, it seems that the code doesn't mention anything about backpropogation. Does pytorch do it automatically? Thanks!