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

update

Browse source
This commit is contained in:
olasaadi 2022-05-30 11:52:47 +03:00
parent 023f8764da
commit 8de77f9afd
1 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -152,9 +152,17 @@ class PyTorchClassifierWrapper(ArtPyTorchClassifier):
self._optimizer.load_state_dict(checkpoint['opt_state_dict'])
def load_latest_checkpoint(self):
"""
Load model only based on the check point path (latest.tar)
:return: loaded model
"""
self.load_checkpoint_by_path('latest.tar')
def load_best_checkpoint(self):
"""
Load model only based on the check point path (model_best.tar)
:return: loaded model
"""
self.load_checkpoint_by_path('model_best.tar')