torch.no_grad(), cleanup

2022-06-28 12:16:44 -04:00
parent 1e2649148b
commit aef24ea157
7 changed files with 24 additions and 115 deletions
@@ -1,6 +1,7 @@
 import torch
 from torch import Tensor
 from torch.nn import Module, ModuleList, GroupNorm, Conv2d, Embedding
+torch.no_grad()

 BATCH_COUNT: int = 1