fix

PFCCLab · Oct 10, 2023 · df47e57 · df47e57
1 parent 5068b29
commit df47e57
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/resnet_dynamo.py b/examples/resnet_dynamo.py
@@ -17,6 +17,7 @@
 
 compiler = TVMCompiler(full_graph=True, print_tabular_mode="rich")
 net = resnet18(pretrained=True, num_classes=2)
+net.eval()  # from_paddle have no arg to set training mode.
 optimized_net = paddlefx.optimize(net, backend=compiler)
 
 x = paddle.rand([1, 3, 224, 224], dtype="float32")