zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-09-01 04:14:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

typo / nits

Browse Source
This commit is contained in:
Awni Hannun
2023-12-14 12:14:01 -08:00
parent b1b9b11801
commit b9439ce74e
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cifar/main.py
View File

@@ -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