Skip to content

Commit d472afa

Browse files
committed
Update mnist_forward_forward function to use --no_accel flag instead of --no_mps --no_cuda
Signed-off-by: jafraustro <jaime.fraustro.valdez@intel.com>
1 parent 1a01f03 commit d472afa

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

run_python_examples.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@ function mnist() {
9393
uv run main.py --epochs 1 --dry-run || error "mnist example failed"
9494
}
9595
function mnist_forward_forward() {
96-
uv run main.py --epochs 1 --no_mps --no_cuda || error "mnist forward forward failed"
96+
uv run main.py --epochs 1 --no_accel || error "mnist forward forward failed"
9797

9898
}
9999
function mnist_hogwild() {

0 commit comments

Comments
 (0)