apunkt/ai-privacy-toolkit
1
0
Fork 0
mirror of https://github.com/IBM/ai-privacy-toolkit.git synced 2026-06-02 14:45:13 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix

Browse source
This commit is contained in:
olasaadi 2022-07-20 17:36:00 +03:00
parent 4973fbebc6
commit 3bf26b67d2
2 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
apt/utils/models/pytorch_model.py
View file

@ -122,6 +122,7 @@ class PyTorchClassifierWrapper(ArtPyTorchClassifier):
tot_correct += correct
total += o_batch.shape[0]
val_loss, val_acc = self._eval(x, y, num_batch, batch_size)
# print acc TODO
best_acc = max(val_acc, best_acc)
if save_checkpoints:
if save_entire_model: