Merge branch 'main' into shape_tutorial

Author: Svetlana Karslioglu

.jenkins/validate_tutorials_built.py

@@ -49,7 +49,6 @@
     "recipes_source/recipes/Captum_Recipe",
     "intermediate_source/flask_rest_api_tutorial",
     "intermediate_source/text_to_speech_with_torchaudio",
-    "intermediate_source/ax_multiobjective_nas_tutorial",
     "intermediate_source/tensorboard_profiler_tutorial" # reenable after 2.0 release.
 ]
 

intermediate_source/mnist_train_nas.py

@@ -44,7 +44,6 @@ def parse_args():
 args = parse_args()
 
 PATH_DATASETS = os.environ.get("PATH_DATASETS", ".")
-AVAIL_GPUS = min(1, torch.cuda.device_count())
 
 
 class MnistModel(LightningModule):
@@ -136,7 +135,6 @@ def run_training_job():
     # Initialize a trainer (don't log anything since things get so slow...)
     trainer = Trainer(
         logger=False,
-        gpus=AVAIL_GPUS,
         max_epochs=args.epochs,
         enable_progress_bar=False,
         deterministic=True,  # Do we want a bit of noise?

intermediate_source/reinforcement_ppo.py

@@ -604,7 +604,7 @@
         data_view = tensordict_data.reshape(-1)
         replay_buffer.extend(data_view.cpu())
         for _ in range(frames_per_batch // sub_batch_size):
-            subdata, *_ = replay_buffer.sample(sub_batch_size)
+            subdata = replay_buffer.sample(sub_batch_size)
             loss_vals = loss_module(subdata.to(device))
             loss_value = (
                 loss_vals["loss_objective"]