imagenet: fix typo addressing args.gpu (#1361)

dvrogozh · web-flow · commit 16554e58b372 · 2025-06-30T11:52:25.000-07:00
diff --git a/imagenet/main.py b/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:
diff --git a/run_python_examples.sh b/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() {

Original file line number	Diff line number	Diff line change
`@@ -81,6 +81,7 @@ function imagenet() {`
`81`	`81`	`cp sample/train/n/* sample/val/n/`
`82`	`82`	`fi`
`83`	`83`	`uv run main.py --epochs 1 sample/ \|\| error "imagenet example failed"`
	`84`	`+ uv run main.py --epochs 1 --gpu 0 sample/ \|\| error "imagenet example failed"`
`84`	`85`	`}`
`85`	`86`
`86`	`87`	`function language_translation() {`