pytorch-lr-scheduler
pytorch-lr-scheduler copied to clipboard
fix some logical errors
I think we should traverse tree path of n2 instead of n1. I have tested on several graphs with LOWMEM_HSM = 1 and LOWMEM_HSM = 0 and found this error.