Skip to content
Snippets Groups Projects

Checkpointing cleanup

Merged Checkpointing cleanup
checkpointing-cleanup into main
Merged Özgür Güler requested to merge checkpointing-cleanup into main
Compare
and
3 files
+ 8
− 12
Compare changes
  • Side-by-side
  • Inline
Files
3
src/ptbench/scripts/predict.py
+ 1
− 1
@@ -131,7 +131,7 @@ def predict(
@@ -131,7 +131,7 @@ def predict(
weight_fullpath = os.path.abspath(weight)
weight_fullpath = os.path.abspath(weight)
checkpointer = Checkpointer(model)
checkpointer = Checkpointer(model)
checkpointer.load(weight_fullpath, strict=False)
checkpointer.load(weight_fullpath)
# Logistic regressor weights
# Logistic regressor weights
if model.name == "logistic_regression":
if model.name == "logistic_regression":