Skip to content

Commit

Permalink
(#2) Victim: fix cifar10 accuracy metric
Browse files Browse the repository at this point in the history
  • Loading branch information
betarixm committed Mar 30, 2022
1 parent ff32b54 commit c127542
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion src/victim/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,6 @@ def train_cifar10_classifier(epochs: int = 100):
test_set,
(32, 32, 3),
name="victim_classifier_cifar10",
accuracy=keras.metrics.CategoricalAccuracy(name="accuracy"),
)
reformer.train(epochs)

Expand Down

0 comments on commit c127542

Please sign in to comment.