From 4f99f9b4428203c392a65c78d57d74ee140112d7 Mon Sep 17 00:00:00 2001 From: Tarun Karuturi Date: Mon, 7 Oct 2024 15:35:52 -0700 Subject: [PATCH] Fix missing export_for_training import in bundled io tutorial (#5949) Summary: Add missing `export_for_training` import in bundledio tutorial. Pull Request resolved: https://github.com/pytorch/executorch/pull/5949 Reviewed By: Olivia-liu, dvorjackz Differential Revision: D63951201 fbshipit-source-id: 42648d6d3e09edfd77714cf4a474d64dcd6c5e07 (cherry picked from commit 37a13976d2ab6c6b5c1fd640dde7d2ee7f688f26) --- docs/source/bundled-io.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/source/bundled-io.md b/docs/source/bundled-io.md index 776c37a5da3..74c86cb8cc7 100644 --- a/docs/source/bundled-io.md +++ b/docs/source/bundled-io.md @@ -313,7 +313,7 @@ from executorch.exir import to_edge from executorch.devtools import BundledProgram from executorch.devtools.bundled_program.config import MethodTestCase, MethodTestSuite -from torch.export import export +from torch.export import export, export_for_training class Module(torch.nn.Module): @@ -449,7 +449,7 @@ from executorch.exir import to_edge from executorch.devtools import BundledProgram from executorch.devtools.bundled_program.config import MethodTestCase, MethodTestSuite -from torch.export import export +from torch.export import export, export_for_training class Module(torch.nn.Module):