Hi, is there any update on this?
I ran it again on my end with pytorch 1.7.1 and above, and I didn’t get the bug:
Before all: True
0 Before: False Inside (decorator): False After: False
1 Before: False Inside (decorator): False After: False
2 Before: False Inside (decorator): False After: False
3 Before: False Inside (decorator): False After: False
4 Before: False Inside (decorator): False After: False
Before all: True
0 Before: False Inside (with): False After: False
1 Before: False Inside (with): False After: False
2 Before: False Inside (with): False After: False
3 Before: False Inside (with): False After: False
4 Before: False Inside (with): False After: False
Before all: True
0 Before: True Inside (decorator): False After: True
1 Before: True Inside (decorator): False After: True
2 Before: True Inside (decorator): False After: True
3 Before: True Inside (decorator): False After: True
4 Before: True Inside (decorator): False After: True
Before all: True
0 Before: True Inside (with): False After: True
1 Before: True Inside (with): False After: True
2 Before: True Inside (with): False After: True
3 Before: True Inside (with): False After: True
4 Before: True Inside (with): False After: True