diff --git a/cifar/README.md b/cifar/README.md index 2016200df..734da7ab7 100644 --- a/cifar/README.md +++ b/cifar/README.md @@ -60,5 +60,5 @@ $ cat >hostfile.json {"ssh": "host-to-ssh-to", "ips": ["ip-to-bind-to"]}, {"ssh": "host-to-ssh-to", "ips": ["ip-to-bind-to"]} ] -$ mlx.launch --verbose --hostfile hostfile.json main.py --batch 256 --epochs 5 --arch resnet20 +$ mlx.launch --verbose --hostfile hostfile.json main.py --batch_size 256 --epochs 5 --arch resnet20 ``` diff --git a/cvae/README.md b/cvae/README.md index 8cfa26ae0..2e1620917 100644 --- a/cvae/README.md +++ b/cvae/README.md @@ -25,7 +25,7 @@ To see the supported options, do `python main.py -h`. Training with the default options should give: ```shell -$ python train.py +$ python main.py Options: Device: GPU Seed: 0