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 @@
 from typing import List
 import torch
 from torch import nn, BoolTensor, FloatTensor, LongTensor
+torch.no_grad()


 class GLUTorch(nn.Module):