Skip to content

Commit 37a1397

Browse files
tarun292facebook-github-bot
authored andcommitted
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
1 parent c86d0d0 commit 37a1397

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)