imagenet: fix typo addressing args.gpu

Fixes: dcc2474 ("Add Accelerator Api to Imagenet Example") Signed-off-by: Dmitry Rogozhkin <dmitry.v.rogozhkin@intel.com>

Author: dvrogozh

imagenet/main.py

@@ -372,7 +372,7 @@ def run_validate(loader, base_progress=0):
  i = base_progress + i
  if use_accel:
  if args.gpu is not None and device.type=='cuda':
- torch.accelerator.set_device_index(argps.gpu)
+ torch.accelerator.set_device_index(args.gpu)
  images = images.cuda(args.gpu, non_blocking=True)
  target = target.cuda(args.gpu, non_blocking=True)
  else:

run_python_examples.sh

@@ -81,6 +81,7 @@ function imagenet() {
  cp sample/train/n/* sample/val/n/
  fi
  uv run main.py --epochs 1 sample/ || error "imagenet example failed"
+ uv run main.py --epochs 1 --gpu 0 sample/ || error "imagenet example failed"
 }
 
 function language_translation() {