diff --git a/cifar/README.md b/cifar/README.md index 118aef9e..d6bdaf9a 100644 --- a/cifar/README.md +++ b/cifar/README.md @@ -47,5 +47,5 @@ Epoch: 99 | Test acc 0.807 Note this was run on an M1 Macbook Pro with 16GB RAM. At the time of writing, `mlx` doesn't have built-in learning rate schedules, -nor a `BatchNorm` layer. We intend to update this example once these features +or a `BatchNorm` layer. We intend to update this example once these features are added. diff --git a/cifar/main.py b/cifar/main.py index 26d06a6a..829417b1 100644 --- a/cifar/main.py +++ b/cifar/main.py @@ -65,7 +65,7 @@ def train_epoch(model, train_iter, optimizer, epoch): ) ) - eean_tr_loss = mx.mean(mx.array(losses)) + mean_tr_loss = mx.mean(mx.array(losses)) mean_tr_acc = mx.mean(mx.array(accs)) samples_per_sec = mx.mean(mx.array(samples_per_sec)) return mean_tr_loss, mean_tr_acc, samples_per_sec