Skip to content

Commit ee07d78

Browse files
pytorchbottarun292
andauthored
Fix missing export_for_training import in bundled io tutorial (#5958)
Fix missing export_for_training import in bundled io tutorial (#5949) Summary: Add missing `export_for_training` import in bundledio tutorial. Pull Request resolved: #5949 Reviewed By: Olivia-liu, dvorjackz Differential Revision: D63951201 fbshipit-source-id: 42648d6d3e09edfd77714cf4a474d64dcd6c5e07 (cherry picked from commit 37a1397) Co-authored-by: Tarun Karuturi <[email protected]>
1 parent 19e5a45 commit ee07d78

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docs/source/bundled-io.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -313,7 +313,7 @@ from executorch.exir import to_edge
313313
from executorch.devtools import BundledProgram
314314
315315
from executorch.devtools.bundled_program.config import MethodTestCase, MethodTestSuite
316-
from torch.export import export
316+
from torch.export import export, export_for_training
317317
318318
319319
class Module(torch.nn.Module):
@@ -449,7 +449,7 @@ from executorch.exir import to_edge
449449
from executorch.devtools import BundledProgram
450450

451451
from executorch.devtools.bundled_program.config import MethodTestCase, MethodTestSuite
452-
from torch.export import export
452+
from torch.export import export, export_for_training
453453

454454

455455
class Module(torch.nn.Module):

0 commit comments

Comments
 (0)