with torch.no_grad()
By Jimmy Upton at Jul 14 2020
with torch.no_grad() if condition
By Jimmy Upton at Sep 13 2020