Skip to content

Commit ebce103

Browse files
BalandatesantorellaSvetlana Karslioglumalfet
authored
Add new tutorial: Multi-Objective NAS with Ax (#2006)
* Add Ax multi-objective NAS optimization tutorial * F Fix reading local file path. Add nbformat to reqs * Use sphinx-gallery built-in support for plotly * Remove weights_summary arg from Trainer (removed in ptl 1.7.0) * changes as recommended in task + pin lightning * Fixed broken links and rst warnings; changed CV figure to plotly; added nbformat as dependency; removed weights_format argument to allow newer version of lightning * Reverted changes recommended in task -- total_trials and downloading data in Makefile * a single space * Various updates * Apply editorial edit suggestions from code review Co-authored-by: Svetlana Karslioglu <svekars@fb.com> * Apply suggestions from code review * Use python builtin component to simplify things * Apply suggestions from code review Co-authored-by: Nikita Shulga <nikita.shulga@gmail.com> * Refactor main loop and guard by `__main__`, print to stdout * Mark mnist_train_nas.py as dependency for tutorial * Move tutorial to intermediate directory Co-authored-by: Elizabeth Santorella <santorella@fb.com> Co-authored-by: Svetlana Karslioglu <svekars@fb.com> Co-authored-by: Nikita Shulga <nikita.shulga@gmail.com> Co-authored-by: Nikita Shulga <nshulga@fb.com>
1 parent 8b8e5d9 commit ebce103

File tree

7 files changed

+709
-2
lines changed

7 files changed

+709
-2
lines changed

Diff for: .jenkins/metadata.json

+3
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,7 @@
11
{
2+
"intermediate_source/ax_multiobjective_nas_tutorial.py": {
3+
"extra_files": ["intermediate_source/mnist_train_nas.py"]
4+
},
25
"beginner_source/dcgan_faces_tutorial.py": {
36
"duration": 2000
47
},

Diff for: _static/img/ax_scheduler_illustration.png

185 KB
Loading

Diff for: conf.py

+3
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,9 @@
3838
import distutils.file_util
3939
import re
4040

41+
import plotly.io as pio
42+
pio.renderers.default = 'sphinx_gallery'
43+
4144

4245
try:
4346
import torchvision

Diff for: index.rst

+10-2
Original file line numberDiff line numberDiff line change
@@ -492,7 +492,7 @@ What's new in PyTorch tutorials?
492492
:image: _static/img/thumbnails/cropped/60-min-blitz.png
493493
:link: advanced/static_quantization_tutorial.html
494494
:tags: Quantization
495-
495+
496496
.. customcarditem::
497497
:header: Grokking PyTorch Intel CPU Performance from First Principles
498498
:card_description: A case study on the TorchServe inference framework optimized with Intel® Extension for PyTorch.
@@ -507,6 +507,13 @@ What's new in PyTorch tutorials?
507507
:link: intermediate/nvfuser_intro_tutorial.html
508508
:tags: Model-Optimization
509509

510+
.. customcarditem::
511+
:header: Multi-Objective Neural Architecture Search with Ax
512+
:card_description: Learn how to use Ax to search over architectures find optimal tradeoffs between accuracy and latency.
513+
:image: _static/img/ray-tune.png
514+
:link: intermediate/ax_multiobjective_nas_tutorial.html
515+
:tags: Model-Optimization,Best-Practice,Ax,TorchX
516+
510517
.. Parallel-and-Distributed-Training
511518
512519
.. customcarditem::
@@ -871,6 +878,7 @@ Additional Resources
871878
advanced/static_quantization_tutorial
872879
intermediate/torchserve_with_ipex
873880
intermediate/nvfuser_intro_tutorial
881+
intermediate/ax_multiobjective_nas_tutorial
874882

875883
.. toctree::
876884
:maxdepth: 2
@@ -902,7 +910,7 @@ Additional Resources
902910

903911
beginner/deeplabv3_on_ios
904912
beginner/deeplabv3_on_android
905-
913+
906914
.. toctree::
907915
:maxdepth: 2
908916
:includehidden:

0 commit comments

Comments
 (0)