wico-silva
commited on
Commit
·
f00f5bc
1
Parent(s):
53d1ae9
fix(tools): disable autograd during TensorRT conversion (#1118)
Browse files- tools/trt.py +1 -0
tools/trt.py
CHANGED
|
@@ -35,6 +35,7 @@ def make_parser():
|
|
| 35 |
|
| 36 |
|
| 37 |
@logger.catch
|
|
|
|
| 38 |
def main():
|
| 39 |
args = make_parser().parse_args()
|
| 40 |
exp = get_exp(args.exp_file, args.name)
|
|
|
|
| 35 |
|
| 36 |
|
| 37 |
@logger.catch
|
| 38 |
+
@torch.no_grad()
|
| 39 |
def main():
|
| 40 |
args = make_parser().parse_args()
|
| 41 |
exp = get_exp(args.exp_file, args.name)
|